fix: adds translations

app.py

@@ -2,12 +2,20 @@ import gradio as gr
 from transformers import pipeline
 
 model_id = "Salesforce/blip-image-captioning-large"
+translate_model_id = "facebook/nllb-200-distilled-600M"
+
 caption_eng = pipeline(model=model_id)
 
 def evaluate(input):
-    return caption_eng(input)[0]["generated_text"]
+    eng = caption_eng(input)[0]["generated_text"]
+    nob = translate(eng)
+    return {eng: eng, nob: nob}
+
+def translate(text, source='eng_Latn', target='nob_Latn'):
+    translator = pipeline( model=translate_model_id, src_lang=source, tgt_lang=target)
+    return translator(text, max_length=400)
 
-iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs="text")
+iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs="json")
 
 if __name__ == "__main__":
     iface.launch()