Update app.py

app.py

@@ -68,19 +68,21 @@ print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
 pipe = pipe.to(device)
 
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
+
+model = model.to(torch.device(device))
 model.eval() #.half()
 
 
 weightsPATH = './clipseg/weights/rd64-uni.pth'
 
-model1 = torch.load(weightsPATH)
-print ("Torch load : ", model1)
-
 state = {'model': model.state_dict()}
 torch.save(state, weightsPATH)
 
 model.load_state_dict(torch.load(weightsPATH['model'], map_location=torch.device(device)), strict=False) #False
 #model.load_state_dict(torch.load(weightsPATH)['model'])
+
+print ("Torch load(model) : ", model)
+print ("Weights : ")
 # print weights
 for k, v in model.named_parameters():
     print(k, v)