Hugging Face's logo

Spaces:
PenguinMan
/
News_Bias_Discrminator
Runtime error

App Files Community
PenguinMan commited on
Commit
5000e4e
·
1 Parent(s): a2b4655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ class SentimentClassifier(nn.Module):
32
  model = SentimentClassifier(len(class_names))
33
 
34
 
35
- device = torch.device('cpu')
36
- model2 = torch.load("model_BERT_2", map_location=device)
37
  def result_final(new_article):
38
 
39
  encoded_review = tokenizer.encode_plus(
 
32
  model = SentimentClassifier(len(class_names))
33
 
34
 
35
+
36
+ model2 = torch.load("model_BERT_2", map_location=torch.device('cpu'))
37
  def result_final(new_article):
38
 
39
  encoded_review = tokenizer.encode_plus(