awakenai commited on
Commit
3ae8602
1 Parent(s): c5439fa
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ app = FastAPI()
9
 
10
  #generator = pipeline('text-generation',model='gpt2')
11
  #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
12
- generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
13
 
14
 
15
  @app.get("/")
 
9
 
10
  #generator = pipeline('text-generation',model='gpt2')
11
  #generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
12
+ #generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
13
 
14
 
15
  @app.get("/")