Update main.py (#3)

main.py

@@ -8,7 +8,9 @@ from fastapi import FastAPI
 app = FastAPI()
 
 #generator = pipeline('text-generation',model='gpt2')
-generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
+#generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
+generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
+
 
 @app.get("/")
 async def root():