Spaces:

Neu256
/

PromeMobile

Sleeping

Neu256 commited on 21 days ago

Commit

f8e8d24

verified ·

1 Parent(s): 488f6f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ model = Transformer(
 m = model.to(DEVICE)
 # print the number of parameters in the model
-m.load_state_dict(torch.load("base_model_1.pth", map_location=torch.device(DEVICE)))
 m.eval()
 #print(

 m = model.to(DEVICE)
 # print the number of parameters in the model
+m.load_state_dict(torch.load("base_model.pth", map_location=torch.device(DEVICE)))
 m.eval()
 #print(