Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ print(device)
|
|
11 |
|
12 |
app = FastAPI()
|
13 |
|
14 |
-
modelName = "
|
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 |
|