Spaces:

BigData-KSU
/

Glaucoma-Detection

Sleeping

bigmed@bigmed commited on Mar 20, 2023

Commit

edae4b7

1 Parent(s): e0fd7bf

added load map location cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ def Final_Compute_regression_results_Sample(Model, batch_sampler,num_head=2):
 #load model
 DeepLab=Transformer_Regression(image_dim=image_shape,dim_patch=dim_patch,num_classes=3,scale=scale,feat_dim=128)
 DeepLab.to(device=device)
-DeepLab.load_state_dict(torch.load("TrainAll_Maghrabi84_50iteration_SWIN.pth.tar"))
 def infer(img):
     # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

 #load model
 DeepLab=Transformer_Regression(image_dim=image_shape,dim_patch=dim_patch,num_classes=3,scale=scale,feat_dim=128)
 DeepLab.to(device=device)
+DeepLab.load_state_dict(torch.load("TrainAll_Maghrabi84_50iteration_SWIN.pth.tar", map_location=torch.device(device)))
 def infer(img):
     # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)