Update app.py

app.py

@@ -7,22 +7,17 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-textgen = load_tool(
-    "TheBloke/Mistral-7B-Instruct-v0.1-GGU",  # replace with a real Hugging Face repo
-    trust_remote_code=True
+model = HfApiModel(
+    model_id="TheBloke/Mistral-7B-Instruct-v0.1-GGUF",  # public model
+    max_tokens=1024,
+    temperature=0.7
 )
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
 @tool
 def ask_question(question: str) -> str:
-    """
-    Ask any general question and get an AI-generated answer.
-    
-    Args:
-        question: The question you want to ask the AI
-    """
-    # Call the loaded textgen tool with the question
-    return textgen(question)
+    """Ask a general question using Mistral"""
+    return model(question)
     
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type