iapain commited on
Commit
e161a0b
1 Parent(s): 2b622c4

fix: outputs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 {'eng': eng, 'nob': 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="json")
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()