Upload main.py

main.py

@@ -14,13 +14,13 @@ embeddings.load('index')
 # Create extractor instance
 extractor = Extractor(embeddings, "google/flan-t5-base")
 
-pipe = pipeline("text2text-generation", model="google/flan-t5-large")
+pipe = pipeline("text2text-generation", model="google/flan-t5-xxl")
 
 
 @app.get("/generate")
 def generate(text: str):
     """
-    deployed flan-t5-large model as backend
+    deployed flan-t5-xxl model as backend
     """
     output = pipe(text)
     return {"output": output[0]["generated_text"]}