Update app.py

app.py

@@ -156,17 +156,12 @@ class LocalModel:
         )
 
     def generate(self, prompt, **kwargs):
-        """
-        Generate text from a given prompt.
-        Args:
-            prompt (str): Input prompt for generation.
-            **kwargs: Additional parameters for the pipeline.
-        Returns:
-            str: Generated text output.
-        """
         result = self.pipeline(prompt, max_new_tokens=500, do_sample=True, **kwargs)
         return result[0]['generated_text']
 
+    def __call__(self, prompt, **kwargs):
+        return self.generate(prompt, **kwargs)
+
 # -------------------- Agent Setup --------------------
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()