Update app.py

app.py

@@ -22,7 +22,8 @@ def validate_images(img_url_1, img_url_2, img_url_3, domain: str, title: str, th
     domain_text = domain.replace("_", " ").lower()
     if title == "":
         title = None
-    probas, valid_probas, invalid_probas = detector.predict_probas_url(img_urls, domain_text, site, title)
+    images, output = detector.predict_probas_url(img_urls, domain_text, site, title)
+    probas, valid_probas, invalid_probas = output
     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