zArabi commited on
Commit
dcf0a46
1 Parent(s): 3438ae8

HooshvareLab-bert-fa-base-uncased-3class-best-epoch-weight-decay=.001 > parsbert-final-v1(epoch3)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ config = BertConfig.from_pretrained(
20
  label2id=label2id)
21
 
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
- path="HooshvareLab-bert-fa-base-uncased-3class-best-epoch-weight-decay=.001.bin"
24
  downloadedModel = hf_hub_download(repo_id="zArabi/Persian-Sentiment-Analysis", filename=path)
25
  loaded_model = torch.load(downloadedModel,map_location=device)
26
  loaded_model.eval()
 
20
  label2id=label2id)
21
 
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
+ path="parsbert-final-v1(epoch3).bin"
24
  downloadedModel = hf_hub_download(repo_id="zArabi/Persian-Sentiment-Analysis", filename=path)
25
  loaded_model = torch.load(downloadedModel,map_location=device)
26
  loaded_model.eval()