Spaces:

FFZG-cleopatra
/

Croatian-News-Sentiment-Classifier-V1

Runtime error

thak123 commited on May 10, 2023

Commit

a7e030a

•

1 Parent(s): 7725e91

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,8 @@ from huggingface_hub import hf_hub_download,snapshot_download
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",filename = "pytorch_model.bin")
 multitask_model.load_state_dict(torch.load(model_link, map_location=device))
 multitask_model.to(device)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# Version 1 -  Croatian Document + Slovenian Document.
+model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-Document-News-Sentiment-Classifier",filename = "pytorch_model.bin")
 multitask_model.load_state_dict(torch.load(model_link, map_location=device))
 multitask_model.to(device)