Update main.py

main.py

@@ -9,7 +9,8 @@ app = FastAPI()
 
 #generator = pipeline('text-generation',model='gpt2')
 #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
-generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
+generator = pipeline("text-generation", model="TheBloke/zephyr-7B-alpha-GGUF")
+#model = AutoModel.from_pretrained("TheBloke/zephyr-7B-alpha-GGUF")
 
 
 @app.get("/")