Hugging Face's logo

Spaces:
trnt
/
twitter_emotions
Running

App Files Community
trnt commited on
Commit
abbb908
•
1 Parent(s): 5653972

using cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer
5
  from transformers import pipeline
6
 
7
  model_path = "trnt/twitter_emotions"
8
- is_gpu = True
9
  device = torch.device('cuda') if is_gpu else torch.device('cpu')
10
  print(device)
11
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
 
5
  from transformers import pipeline
6
 
7
  model_path = "trnt/twitter_emotions"
8
+ is_gpu = False
9
  device = torch.device('cuda') if is_gpu else torch.device('cpu')
10
  print(device)
11
  model = AutoModelForSequenceClassification.from_pretrained(model_path)