Update app.py

app.py

@@ -23,13 +23,15 @@ class SmartAgent:
     def __init__(self):
         self.generator = pipeline(
             "text2text-generation",
-            model="google/flan-t5-base",
+            model="google/flan-ul2",
+            torch_dtype="auto",
             max_new_tokens=128,
             temperature=0.3,
             do_sample=True,
             device_map="auto"
         )
 
+
         # self.system_prompt = (
         #     "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer "
         #     "with the following template: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR "