Haseeb javed commited on
Commit
cddf948
·
1 Parent(s): a58a3ee
Files changed (1) hide show
  1. app.py +5 -1
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