Spaces:

IBounhas
/

Riadh

Runtime error

IBounhas commited on Nov 17, 2023

Commit

acb7c8e

1 Parent(s): 1826c87

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ class BertForSTS(torch.nn.Module):
         #x = self.fc2(x)
         return x
-model_load_path = "IBounhas/riadh/bert-sts-15.pt"
 model = BertForSTS()
 model.load_state_dict(torch.load(model_load_path, map_location=torch.device('cpu')))
 model.to(device)

         #x = self.fc2(x)
         return x
+model_load_path = "https://drive.google.com/file/d/1KaFaeoMAkRcJdhXe77ibmwqz6pseAqkj/view?usp=sharing"
 model = BertForSTS()
 model.load_state_dict(torch.load(model_load_path, map_location=torch.device('cpu')))
 model.to(device)