Hugging Face's logo

Spaces:
Keetawan
/
RobertaEmojiSentimentAnalysis
Sleeping

App Files Community
Keetawan commited on
Commit
c0d0a2f
·
1 Parent(s): 81ef0fe

change: model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from model import SentimentAnalysisModel
5
 
6
  # Load the pre-trained sentiment analysis model
7
  model = SentimentAnalysisModel(bert_model_name="SamLowe/roberta-base-go_emotions", num_labels=7)
8
- model.load_state_dict(torch.load("best_model_0.75.pth", map_location=torch.device('cpu')))
9
  model.eval()
10
 
11
  # Mapping from predicted class to emoji
 
5
 
6
  # Load the pre-trained sentiment analysis model
7
  model = SentimentAnalysisModel(bert_model_name="SamLowe/roberta-base-go_emotions", num_labels=7)
8
+ model.load_state_dict(torch.load("best_model_75.pth", map_location=torch.device('cpu')))
9
  model.eval()
10
 
11
  # Mapping from predicted class to emoji