hf cuda issue

app.py

@@ -473,7 +473,10 @@ def visualize_ref(ref, ex_mask):
            inpaint_mask, opts.image_size, interpolation=cv2.INTER_AREA
         )
         inpaint_mask = (inpaint_mask > 128).astype(np.uint8)
-        img = cv2.cvtColor(ref["background"], cv2.COLOR_RGBA2RGB)
+        try:
+            img = cv2.cvtColor(ref["background"], cv2.COLOR_RGBA2RGB)
+        except:
+            gr.Warning("Internal Error. Please do Clear and try again.", duration=10)
     else:
         inpaint_mask = np.array(ref["layers"][0])[..., -1]
         inpaint_mask = cv2.resize(
@@ -485,6 +488,8 @@ def visualize_ref(ref, ex_mask):
 
     # viualization
     mask = inpainted < 128
+    if img.shape[0] != mask.shape[0]:
+        gr.Warning("Internal Error. Please do Clear and try again.", duration=10)
     img = mask_image(img, mask)
     if inpaint_mask.sum() == 0:
         gr.Warning("Run botton not enabled? Please try again.", duration=10)