Update app.py

app.py

@@ -132,13 +132,13 @@ def get_response(user_input):
     #             device_map='auto'
     # )
 
+    # llm = HuggingFacePipeline.from_model_id(
+    #     model_id="google/flan-t5-base",
+    #     task="text2text-generation",
+    #     # model_kwargs={"temperature": 0.2},
+    # )
     llm = HuggingFacePipeline.from_model_id(
-        model_id="google/flan-t5-base",
-        task="text2text-generation",
-        # model_kwargs={"temperature": 0.2},
-    )
-    llm = HuggingFacePipeline.from_model_id(
-        model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
+        model_id="google-t5/t5-small",
         task="text2text-generation",
         # model_kwargs={"temperature": 0.2},
     )