Update app.py

app.py

@@ -18,15 +18,14 @@ config = BertConfig.from_pretrained(
     id2label=id2label,
     label2id=label2id)
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 path="HooshvareLab-bert-fa-base-uncased-3class-best-epoch-weight-decay=.001.bin"
-downloadedModelFile = hf_hub_download(repo_id="zArabi/Persian-Sentiment-Analysis", filename=path)
-loaded_model = torch.load(downloadedModelFile,map_location="cpu")
+downloadedModel = hf_hub_download(repo_id="zArabi/Persian-Sentiment-Analysis", filename=path)
+loaded_model = torch.load(downloadedModel,map_location=device)
 loaded_model.eval()
 
-
 tokenizer = BertTokenizer.from_pretrained(modelName)
 max_len=512
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def predict(text):
   text = cleaning(text)