Update app.py

app.py

@@ -61,39 +61,41 @@ async def process_inputs(class_names: List[str], image_url: str):
     High level function that takes in the user inputs and returns the
     classification results as panel objects.
     """
-    main.disabled = True
-    if not image_url:
-        yield "##### ⚠️ Provide an image URL"
-        return
-
-    yield "##### ⚙ Fetching image and running model..."
     try:
-        pil_img = await open_image_url(image_url)
-        img = pn.pane.Image(pil_img, height=400, align="center")
-    except Exception as e:
-        yield f"##### 😔 Something went wrong, please try a different URL!"
-        return
-
-    class_items = class_names.split(",")
-    class_likelihoods = get_similarity_scores(class_items, pil_img)
-
-    # build the results column
-    results = pn.Column("##### 🎉 Here are the results!", img)
-
-    for class_item, class_likelihood in zip(class_items, class_likelihoods):
-        row_label = pn.widgets.StaticText(
-            name=class_item.strip(), value=f"{class_likelihood:.2%}", align="center"
-        )
-        row_bar = pn.indicators.Progress(
-            value=int(class_likelihood * 100),
-            sizing_mode="stretch_width",
-            bar_color="secondary",
-            margin=(0, 10),
-            design=pn.theme.Material,
-        )
-        results.append(pn.Column(row_label, row_bar))
-    main.disabled = False
-    yield results
+        if not image_url:
+            yield "##### ⚠️ Provide an image URL"
+            return
+    
+        yield "##### ⚙ Fetching image and running model..."
+        try:
+            pil_img = await open_image_url(image_url)
+            img = pn.pane.Image(pil_img, height=400, align="center")
+        except Exception as e:
+            yield f"##### 😔 Something went wrong, please try a different URL!"
+            main.disabled = False
+            return
+    
+        class_items = class_names.split(",")
+        class_likelihoods = get_similarity_scores(class_items, pil_img)
+    
+        # build the results column
+        results = pn.Column("##### 🎉 Here are the results!", img)
+    
+        for class_item, class_likelihood in zip(class_items, class_likelihoods):
+            row_label = pn.widgets.StaticText(
+                name=class_item.strip(), value=f"{class_likelihood:.2%}", align="center"
+            )
+            row_bar = pn.indicators.Progress(
+                value=int(class_likelihood * 100),
+                sizing_mode="stretch_width",
+                bar_color="secondary",
+                margin=(0, 10),
+                design=pn.theme.Material,
+            )
+            results.append(pn.Column(row_label, row_bar))
+        yield results
+    finally:
+        main.disabled = False
 
 
 # create widgets