Update app.py

app.py

@@ -131,11 +131,11 @@ def image_generator(prompt: str) -> str:
         return f"Image generation failed: {str(e)}"
 
 # -------------------- Local LLM (Replaces HfApiModel) --------------------
-from smolagents import LocalModel
+from transformers import pipeline
 
-class TransformersModel(LocalModel):
+class LocalModel:
     """
-    Local text generation model wrapper using Hugging Face Transformers.
+    Minimal local model interface compatible with smolagents CodeAgent.
     """
     def __init__(self):
         self.pipeline = pipeline(
@@ -159,7 +159,7 @@ class TransformersModel(LocalModel):
         result = self.pipeline(prompt, max_new_tokens=500, do_sample=True)
         return result[0]['generated_text']
 
-model = TransformersModel()
+model = LocalModel()
 
 # -------------------- Agent Setup --------------------
 final_answer = FinalAnswerTool()