Spaces:

transpolymer
/

Transpolymer2

Running

transpolymer commited on May 4

Commit

a0b466d

verified ·

1 Parent(s): fe93a77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ class TransformerRegressor(nn.Module):
 # Load model
 model = TransformerRegressor()
-model.load_state_dict(torch.load("best_model.pt", map_location=torch.device('cpu')))
 model.eval()
 # Feature Functions

 # Load model
 model = TransformerRegressor()
+model.load_state_dict(torch.load("tranformer_model.pt", map_location=torch.device('cpu')))
 model.eval()
 # Feature Functions