Haseeb javed
commited on
Commit
·
cddf948
1
Parent(s):
a58a3ee
3:43
Browse files
app.py
CHANGED
@@ -34,11 +34,15 @@ def generate_response(prompt):
|
|
34 |
token = outputs.sequences[0, input_length:]
|
35 |
return tokenizer.decode(token, skip_special_tokens=True)
|
36 |
|
|
|
|
|
|
|
|
|
37 |
@app.route("/chat", methods=["POST"])
|
38 |
def chat():
|
39 |
data = request.json
|
40 |
user_input = data.get("message", "")
|
41 |
-
prompt = f"<human>: {user_input}\n<bot>:"
|
42 |
response = generate_response(prompt)
|
43 |
return jsonify({"response": response})
|
44 |
|
|
|
34 |
token = outputs.sequences[0, input_length:]
|
35 |
return tokenizer.decode(token, skip_special_tokens=True)
|
36 |
|
37 |
+
@app.route("/", methods=["GET"])
|
38 |
+
def home():
|
39 |
+
return jsonify({"message": "Flask app is running!"})
|
40 |
+
|
41 |
@app.route("/chat", methods=["POST"])
|
42 |
def chat():
|
43 |
data = request.json
|
44 |
user_input = data.get("message", "")
|
45 |
+
prompt = f"<human>: {user_input}\n<bot>:" # Ensure prompt format matches your model's expectation
|
46 |
response = generate_response(prompt)
|
47 |
return jsonify({"response": response})
|
48 |
|