Update app.py

app.py

@@ -67,9 +67,10 @@ print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
 
 pipe = pipe.to(device)
 
-model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
-model.eval()
-model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(device)), strict=True) #False
+model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
+model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(device)), strict=False) #False
+
+model.eval().half()
 
 imgRes = 256