Update app.py

app.py

@@ -49,8 +49,9 @@ def classify_message(message: str) -> str:
                 "content": message,
             },
         ],
-        model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
     )
+    print(classify.choices[0].message.content.strip())
     return classify.choices[0].message.content.strip()
 
 # Function to extract information based on classification
@@ -77,12 +78,13 @@ def extract_info(message: str, classification: str) -> dict:
                 "content": message,
             },
         ],
-        model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
         response_format={
             "type": "json_object",
             "schema": schema.model_json_schema(),
         },
     )
+    print(json.loads(extract.choices[0].message.content))
     return json.loads(extract.choices[0].message.content)
 
 # Function to make API call
@@ -129,8 +131,9 @@ def interpret_response(user_message: str, classification: str, api_response: dic
                 "content": f"User message: {user_message}\nRequest type: {classification}\nAPI response: {json.dumps(api_response)}",
             },
         ],
-        model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
     )
+    print(interpret.choices[0].message.content)
     return interpret.choices[0].message.content
 
 # Main function to process the user request