last resort try catch

app.py

@@ -46,12 +46,16 @@ def generate_relevancy(
     assert img.dtype == np.uint8
     h, w, c = img.shape
     start = time()
-    grads = ClipWrapper.get_clip_saliency(
-        img=img,
-        text_labels=np.array(labels),
-        prompts=prompts,
-        **saliency_configs[saliency_config](h),
-    )[0]
+    try:
+        grads = ClipWrapper.get_clip_saliency(
+            img=img,
+            text_labels=np.array(labels),
+            prompts=prompts,
+            **saliency_configs[saliency_config](h),
+        )[0]
+    except Exception as e:
+        print(e)
+        return [img]
     print("inference took", float(time() - start))
     if subtract_mean:
         grads -= grads.mean(axis=0)