Spaces:
Running
Running
samueldomdey
commited on
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):
|