Update app.py

app.py

@@ -41,12 +41,13 @@ example_images = sorted(
 
 def predict(
     image: Image.Image,
+    model_repo: str,
     threshold: float = 0.5,
 ):
     # join variant for cache key
-    model, transform = load_model_and_transform(MODEL_REPO)
+    model, transform = load_model_and_transform(model_repo)
     # load labels
-    labels: LabelData = load_labels_hf(MODEL_REPO)
+    labels: LabelData = load_labels_hf(model_repo)
     # preprocess image
     image = preprocess_image(image, (448, 448))
     image = transform(image).unsqueeze(0)
@@ -128,10 +129,10 @@ with gr.Blocks(theme="NoCrypt/miku", analytics_enabled=False, title=TITLE, css=c
                     general = gr.Label(label="General")
 
     with gr.Row():
-        examples = [[imgpath, 0.35] for imgpath in example_images]
+        examples = [[imgpath, MODEL_REPO, 0.35] for imgpath in example_images]
         examples = gr.Examples(
             examples=examples,
-            inputs=[img_input, threshold],
+            inputs=[img_input, model_to_use, threshold],
         )
 
     # tell clear button which components to clear
@@ -139,7 +140,7 @@ with gr.Blocks(theme="NoCrypt/miku", analytics_enabled=False, title=TITLE, css=c
 
     submit.click(
         predict,
-        inputs=[img_input, threshold],
+        inputs=[img_input, model_to_use, threshold],
         outputs=[heatmap_gallery, heatmap_grid, caption, tags, rating, character, general],
         api_name="predict",
     )