shubh2014shiv commited on
Commit
2f6ce67
1 Parent(s): 6e9b3f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -145,6 +145,10 @@ if topic == "Sentiment Analysis":
145
  #model.load_state_dict(
146
  # torch.load(JAPANESE_SENTIMENT_PROJECT_PATH + 'FineTuneJapaneseBert_AmazonReviewSentiments.pt',
147
  # map_location=torch.device('cpu')))
 
 
 
 
148
 
149
  inputs = {
150
  'input_ids': input_ids,
 
145
  #model.load_state_dict(
146
  # torch.load(JAPANESE_SENTIMENT_PROJECT_PATH + 'FineTuneJapaneseBert_AmazonReviewSentiments.pt',
147
  # map_location=torch.device('cpu')))
148
+
149
+ model.load_state_dict(
150
+ torch.load('reviewSentiments_jp.pt',
151
+ map_location=torch.device('cpu')))
152
 
153
  inputs = {
154
  'input_ids': input_ids,