Update app.py

app.py

@@ -19,10 +19,10 @@ 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()
+    domain_text = domain.replace("_", " ").lower()
     if title == "":
         title = None
-    images, domain, probas, valid_probas, invalid_probas = detector.predict_probas_url(img_urls, domain, site, title)
+    probas, valid_probas, invalid_probas = detector.predict_probas_url(img_urls, domain_text, 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