Spaces:

transpolymer
/

Transpolymer2

Sleeping

transpolymer commited on May 4

Commit

588c57e

verified ·

1 Parent(s): e555991

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("tranformer_model.pt", map_location=torch.device('cpu')))
 model.eval()
 # Feature Functions

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