Update app.py

app.py

@@ -26,12 +26,18 @@ if "Mistral" not in os.environ:
     os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
 
 print("Loading LLM...")
-chat2 = ChatMistralAI(
+chat3 = ChatMistralAI(
     model="pixtral-large-2411",
     temperature=0.01,
     max_retries=6,
     mistral_api_key = os.getenv("Mistral")
 )
+chat2 = ChatMistralAI(
+    model="mistral-ocr-2505",
+    temperature=0.01,
+    max_retries=6,
+    mistral_api_key = os.getenv("Mistral")
+)
 print(f"Model {chat2.model} downloaded!")
 chat1 = ChatMistralAI(
     model="mistral-medium-2505",
@@ -207,6 +213,9 @@ class BasicAgent:
             except Exception as e:
                 alfred = build_agent(chat2)
                 answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
+            finally:
+                alfred = build_agent(chat3)
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
         if answer:
             answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))    
             answer = answer.replace('FINAL ANSWER:', '')