Spaces:

awakenai
/

fast-api-transformers-pipeline-v2

Paused

awakenai commited on Oct 17, 2023

Commit

694ef91

1 Parent(s): 82f8eb5

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

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

 #generator = pipeline('text-generation',model='gpt2')
 #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
+#generator = pipeline("text-generation", model="TheBloke/zephyr-7B-alpha-GGUF")
 #model = AutoModel.from_pretrained("TheBloke/zephyr-7B-alpha-GGUF")