thak123 commited on
Commit
a7e030a
1 Parent(s): 7725e91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ from huggingface_hub import hf_hub_download,snapshot_download
13
 
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
16
- model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",filename = "pytorch_model.bin")
 
17
 
18
  multitask_model.load_state_dict(torch.load(model_link, map_location=device))
19
  multitask_model.to(device)
 
13
 
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)