scott12355 commited on
Commit
6d4abdf
·
verified ·
1 Parent(s): 886367a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ print(device)
13
 
14
  app = FastAPI()
15
 
16
- modelName = "Qwen/Qwen2.5-1.5B-Instruct" #Qwen/Qwen2.5-1.5B-Instruct
17
  pipe = pipeline("text-generation", model=modelName, device=device, batch_size=8)
18
  sentiment_model = pipeline("sentiment-analysis", device=device)
19
 
 
13
 
14
  app = FastAPI()
15
 
16
+ modelName = "Qwen/Qwen2.5-0.5B-Instruct" #Qwen/Qwen2.5-1.5B-Instruct
17
  pipe = pipeline("text-generation", model=modelName, device=device, batch_size=8)
18
  sentiment_model = pipeline("sentiment-analysis", device=device)
19