Update app.py

app.py

@@ -28,6 +28,8 @@ def predict(img):
         image_tensor=TF.to_tensor(img).cpu()
         image_tensor=image_tensor.unsqueeze(0).cuda()
         ipt=net_c(image_tensor)
+        image_tensor = image_tensor.half()
+        ipt = ipt.half()
         output_i, output_j=net_i(image_tensor,ipt,prompt=True)
         output_j_out=[]
         for i in range(4):