Undid shi

main.py

@@ -1,7 +1,5 @@
 from flask import Flask, request, render_template, jsonify
-from models import mixtral
-
-# https://www.youtube.com/watch?v=irjYqV6EebU
+from models import blenderbot
 
 app = Flask("AI API")
 
@@ -16,14 +14,12 @@ def test_route():
 @app.route("/api", methods=["POST"])
 def receive_data():
   data = request.get_json()
-  
   print("Prompt:", data["prompt"])
-  print("System:", data["system"])
 
-  generated_text = mixtral.generate(data["prompt"], data["system"], data["history"], False, None)
+  generated_text = blenderbot.generate(data["prompt"])
   
   print("Response:", generated_text)
   
-  return { "answer": generated_text }
+  return jsonify({ "answer": generated_text })
 
 app.run(host="0.0.0.0", port=7860, debug=False)