Update app.py

app.py

@@ -112,10 +112,12 @@ def generate_response(user_input, model_id):
         for msg in chat_history:
             func_caller.append({"role": "user", "content": f"{str(msg[0])}"})
             func_caller.append({"role": "assistant", "content": f"{str(msg[1])}"})
+ 
+        highest_label_result = classify_intent(user_input) 
 
         # Reformulated prompt based on intent classification
-        reformulated_prompt = reformulate_prompt(user_input, highest_label)
-        
+        reformulated_prompt = reformulate_prompt(user_input, highest_label_result)
+          
         func_caller.append({"role": "user", "content": f'{reformulated_prompt}'})
         formatted_prompt = "\n".join([f"{m['role']}: {m['content']}" for m in func_caller])
 
@@ -185,7 +187,7 @@ def handle_post_request():
             return jsonify({"error": "No JSON data provided"}), 400
 
         message = data.get("inputs", "No message provided.")
-        model_id = data.get("model_id", "meta-llama/Llama-3.1-8B-Instruct")
+        model_id = data.get("model_id", "openai-community/gpt2-large")
 
         print(f"Processing request with model_id: {model_id}")
         model_response = generate_response(message, model_id)