Spaces:

ldhldh
/

streamlit_demo

Runtime error

ldhldh commited on May 26

Commit

c2da71b

•

1 Parent(s): 4b8188d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = models.get_model(
     config=model_parameters,
     device=device,
 )
-model.load_state_dict(torch.load(model_paths), map_location=torch.device('cpu'))
 model = model.to(device)
 model.eval()

     config=model_parameters,
     device=device,
 )
+model.load_state_dict(torch.load(model_paths, map_location=torch.device('cpu')))
 model = model.to(device)
 model.eval()