samueldomdey commited on
Commit
937bb8a
1 Parent(s): 514ddbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, device=0) # set device=0 to use GPU (CPU default, -1)
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):