Spaces:
Runtime error
Runtime error
Commit
·
937bb8a
1
Parent(s):
514ddbe
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import pandas as pd
|
|
| 5 |
|
| 6 |
# define nlp mask
|
| 7 |
model = "siebert/sentiment-roberta-large-english"
|
| 8 |
-
nlp = pipeline(model=model
|
| 9 |
|
| 10 |
# perform inference on given file
|
| 11 |
def inference(df, filename):
|
|
|
|
| 5 |
|
| 6 |
# define nlp mask
|
| 7 |
model = "siebert/sentiment-roberta-large-english"
|
| 8 |
+
nlp = pipeline(model=model) # set device=0 to use GPU (CPU default, -1)
|
| 9 |
|
| 10 |
# perform inference on given file
|
| 11 |
def inference(df, filename):
|