Update app.py

app.py

@@ -70,7 +70,13 @@ class DocumentRetrievalAndGeneration:
         )
         return generate_text
     def initialize_llm2(self,model_id):
-        pipe = pipeline("text2text-generation", model="google/flan-t5-small")
+        try:
+            pipe = pipeline("text-generation", model="google/gemma-2-9b")
+        except:
+            try:
+                pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
+            except:
+                pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct")
         # pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")   
         # model_name = "mistralai/Mistral-7B-Instruct-v0.2"
         # pipeline = transformers.pipeline(