Spaces:

darshanjani
/

demo0

Sleeping

darshanjani commited on Aug 20, 2023

Commit

0867d1f

•

1 Parent(s): 58c979f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ model = Net(
 model.load_state_dict(
     torch.load(
-        config.MODEL_PATH,
         map_location=torch.device(config.ACCELERATOR)
     )
 )

 model.load_state_dict(
     torch.load(
+        config.MODEL_SAVE_PATH,
         map_location=torch.device(config.ACCELERATOR)
     )
 )