Workaround for HF-spaces render_examples failing with partial examples

app.py

@@ -8,20 +8,24 @@ models = {
 }
 
 def predict(img, model_type):
-    model = models[model_type]
-    results = model(img, size=640)
-    detection_img = results.render()[0]
+    if model_type in models:
+        model = models[model_type]
+        results = model(img, size=640)
+        detection_img = results.render()[0]
 
-    return detection_img
+        return detection_img, "Prediction ready"
+
+    return img, f"Model '{model_type}' is not supported"
 
 # Interface
 inputs = [
     "image",
-    gr.Dropdown(["yolov5s", "yolov5m"], label="Model", value="yolov5s")
+    gr.Dropdown(["yolov5s", "yolov5m"], label="Model", value="yolov5s"),
 ]
 
 outputs = [
-    "image"
+    "image",
+    gr.Textbox(label="Status"),
 ]
 
 examples = [