thak123 commited on
Commit
fb09861
1 Parent(s): f08fa03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import hf_hub_download,snapshot_download
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
16
  # Version 1 - Croatian Document + Slovenian Document.
17
- model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-Document-News-Sentiment-Classifier",filename = "pytorch_model.bin")
18
 
19
  multitask_model.load_state_dict(torch.load(model_link, map_location=device))
20
  multitask_model.to(device)
@@ -69,7 +69,7 @@ interface = gr.Interface(
69
  fn=predict_sentiment,
70
  inputs='text',
71
  outputs=['label'],
72
- title='Croatian News Sentiment Analysis 1.0',
73
  description='Get the positive/neutral/negative sentiment for the given input.'
74
  )
75
 
 
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
16
  # Version 1 - Croatian Document + Slovenian Document.
17
+ model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-Document-News-Sentiment-Classifier-V2",filename = "pytorch_model.bin")
18
 
19
  multitask_model.load_state_dict(torch.load(model_link, map_location=device))
20
  multitask_model.to(device)
 
69
  fn=predict_sentiment,
70
  inputs='text',
71
  outputs=['label'],
72
+ title='Croatian News Sentiment Analysis 2.0',
73
  description='Get the positive/neutral/negative sentiment for the given input.'
74
  )
75