sasha HF staff commited on
Commit
c62d033
1 Parent(s): 5cc1737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ eval = evaluator("text-classification")
60
  ### Loading models
61
 
62
  for i in range (len(models)):
63
- globals()[f"pipe_{i}"] = pipeline("text-classification", model = models[i], tokenizer = models[i], device=0)
64
  st.text("Loaded model "+ str(models[i]))
65
 
66
  ### Defining metrics
 
60
  ### Loading models
61
 
62
  for i in range (len(models)):
63
+ globals()[f"pipe_{i}"] = pipeline("text-classification", model = models[i], tokenizer = models[i], device=-1)
64
  st.text("Loaded model "+ str(models[i]))
65
 
66
  ### Defining metrics