Upload 2 files

app.py

@@ -1,12 +1,16 @@
 from fastapi import FastAPI
 from transformers import pipeline
-
+from transformers import pipeline
 ## create a new FASTAPI app instance
 app=FastAPI()
 
 # Initialize the text generation pipeline
-pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
+#pipe = pipeline("text2text-generation", model="google/flan-t5-small")
+# Use a pipeline as a high-level helper
+
+
+pipe = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1")
 
 @app.get("/")
 def home():