scott12355 commited on
Commit
10f7bb7
·
verified ·
1 Parent(s): 9481a5c

Update app.py

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