Update app.py
Browse files
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
|
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 |
|