Spaces:

Nefflymicn
/

Amazon-Sentiment-LiveDemo

Sleeping

Nefflymicn commited on 3 days ago

Commit

0cdbb60

verified ·

1 Parent(s): d05cfa9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ model = TransformerSentimentModel(
     output_dim=2
 )
 weights_path = hf_hub_download(repo_id=REPO_ID, filename="pytorch_model.bin", revision="v2.0")
-model.load_state_dict(torch.load("pytorch_model.bin", map_location=device))
 model.eval()

     output_dim=2
 )
 weights_path = hf_hub_download(repo_id=REPO_ID, filename="pytorch_model.bin", revision="v2.0")
+model.load_state_dict(torch.load(weights_path, map_location=device))
 model.eval()