fix: outputs
Browse files
app.py
CHANGED
@@ -13,12 +13,12 @@ translator = pipeline(model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt
|
|
13 |
def evaluate(input):
|
14 |
eng = caption_eng(input)[0]["generated_text"]
|
15 |
nob = translate(eng)
|
16 |
-
return
|
17 |
|
18 |
def translate(text):
|
19 |
return translator(text, max_length=400)
|
20 |
|
21 |
-
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs="
|
22 |
|
23 |
if __name__ == "__main__":
|
24 |
iface.launch()
|
|
|
13 |
def evaluate(input):
|
14 |
eng = caption_eng(input)[0]["generated_text"]
|
15 |
nob = translate(eng)
|
16 |
+
return (eng, nob)
|
17 |
|
18 |
def translate(text):
|
19 |
return translator(text, max_length=400)
|
20 |
|
21 |
+
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs=["text", "text"])
|
22 |
|
23 |
if __name__ == "__main__":
|
24 |
iface.launch()
|