Spaces:

ain3007-project
/

model-inference

Sleeping

osbm commited on Jan 17, 2024

Commit

b38da9b

verified ·

1 Parent(s): 8efcd10

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = UNet(
     num_res_units=4,
     dropout=0.15,
 )
-model.load_state_dict(torch.load("best_model.pth"), map_location=torch.device('cpu'))
 model.eval()
 def greet(image_path):

     num_res_units=4,
     dropout=0.15,
 )
+model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
 model.eval()
 def greet(image_path):