Update app.py

app.py

@@ -18,9 +18,11 @@ def validate_item(item_id: str, use_title: bool, threshold: float):
 
 def validate_images(img_url_1, img_url_2, img_url_3, domain: str, title: str, threshold: float):
     img_urls = [url for url in [img_url_1, img_url_2, img_url_3] if url != ""]
+    site, domain = domain.split("-")
+    domain = domain.replace("_", " ").lower()
     if title == "":
         title = None
-    images, domain, probas, valid_probas, invalid_probas = detector.predict_probas_url(img_urls, domain, title)
+    images, domain, probas, valid_probas, invalid_probas = detector.predict_probas_url(img_urls, domain, site, title)
     valid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() >= threshold]
     invalid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() < threshold]
     return f"## Domain: {domain}", valid_images, invalid_images
@@ -60,7 +62,7 @@ with gr.Blocks() as demo:
                 img_url_2 = gr.Textbox(label="Picture URL")
                 img_url_3 = gr.Textbox(label="Picture URL")
             with gr.Column():
-                domain = gr.Textbox(label="Domain name", placeholder="Required")
+                domain = gr.Textbox(label="Domain ID", placeholder="Required")
                 title = gr.Textbox(label="Item title", placeholder="Optional")
                 threshold = gr.Number(label="Threshold", value=0.25, precision=2)
             submit = gr.Button("Submit")