Update off_topic.py

off_topic.py

@@ -116,14 +116,14 @@ class OffTopicDetector:
         invalid_probas = probas[:, n_valid:n_classes].sum(axis=1, keepdims=True)
         return probas, valid_probas, invalid_probas
 
-    def predict_probas_url(self, img_urls: List[str], domain: str,
+    def predict_probas_url(self, img_urls: List[str], domain: str, site:str, 
                            title: Optional[str] = None,
                 valid_templates: Optional[List[str]] = None,
                 invalid_classes: Optional[List[str]] = None,
                 autocast: bool = True):
         images = self.get_images(img_urls)
         dedup_images = self._filter_dups(images)
-        return self.predict_probas(images, domain, title, valid_templates, invalid_classes, autocast)
+        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,