Update main.py

main.py

@@ -6,7 +6,10 @@ from transformers import pipeline
 
 app = FastAPI()
 
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+#pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-large")
+#Try large rather than small?  google/flan-t5-small
+
 
 @app.get("/infer_t5")
 def t5(input):