Update app.py

app.py

@@ -69,7 +69,22 @@ pipe = pipe.to(device)
 
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
 model.eval() #.half()
-model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(device)), strict=False) #False
+
+
+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 weights
+for k, v in model.named_parameters():
+    print(k, v)
+    
 
 imgRes = 256