Spaces:

rxavier
/

off-topic-images

Runtime error

rxavier commited on Aug 23, 2023

Commit

8615061

•

1 Parent(s): 31e18ef

Update off_topic.py

Files changed (1) hide show

off_topic.py CHANGED Viewed

@@ -123,7 +123,7 @@ class OffTopicDetector:
                 autocast: bool = True):
         images = self.get_images(img_urls)
         dedup_images = self._filter_dups(images)
-        return self.predict_probas(images, domain, site, title, valid_templates, invalid_classes, autocast)
     def predict_probas_item(self, url_or_id: str,
                             use_title: bool = False,

                 autocast: bool = True):
         images = self.get_images(img_urls)
         dedup_images = self._filter_dups(images)
+        return dedup_images, self.predict_probas(dedup_images, domain, site, title, valid_templates, invalid_classes, autocast)
     def predict_probas_item(self, url_or_id: str,
                             use_title: bool = False,