Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
 
-from off_topic import OffTopicDetector
+from off_topic import OffTopicDetector, Translator
+
+
+translator = Translator("facebook/nllb-200-distilled-600M")
+detector = OffTopicDetector("openai/clip-vit-base-patch32", image_size="V", translator=translator)
 
-detector = OffTopicDetector("openai/clip-vit-base-patch32")
 
 def validate(item_id: str, use_title: bool, threshold: float):
     images, domain, probas, valid_probas, invalid_probas = detector.predict_probas_item(item_id, use_title=use_title)
@@ -10,6 +13,7 @@ def validate(item_id: str, use_title: bool, threshold: float):
     invalid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() < threshold]
     return f"## Domain: {domain}", valid_images, invalid_images
 
+
 with gr.Blocks() as demo:
     gr.Markdown("""
                 # Off topic image detector