Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ print(device)
|
|
13 |
|
14 |
app = FastAPI()
|
15 |
|
16 |
-
modelName = "Qwen/Qwen2.5-
|
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 |
|