Fuk large models

main.py

@@ -1,5 +1,5 @@
 from flask import Flask, request, render_template, jsonify
-from models import hermes
+from models import blenderbot
 
 app = Flask("AI API")
 
@@ -16,19 +16,7 @@ def receive_data():
   data = request.get_json()
   print("Prompt:", data["prompt"])
   
-  messages = []
-  
-  if data["system"]:
-    messages.append({"role": "system", "content": data["system"] })
-  
-  messages.append(
-    {
-      "role": "user",
-      "content": data["prompt"]
-    }
-  )
-  
-  generated_text = hermes.generate(messages)
+  generated_text = blenderbot.generate(data["prompt"])
   
   print("Response:", generated_text)