Update main.py

main.py

@@ -6,8 +6,8 @@ from transformers import pipeline # pipeline for inference
 
 app = FastAPI()
 
-#pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
-pipe_flan = pipeline("text2text-generation", model="mithril-security/gpt-j-6B")
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+#pipe_flan = pipeline("text2text-generation", model="mithril-security/gpt-j-6B")
 
 @app.get("/infer_t5")
 def t5(input):