Spaces:
Runtime error
Runtime error
fix: ui improvements
Browse files
app.py
CHANGED
@@ -8,17 +8,17 @@ caption_eng = pipeline(model=model_id)
|
|
8 |
|
9 |
model = AutoModelForSeq2SeqLM.from_pretrained(translate_model_id)
|
10 |
tokenizer = AutoTokenizer.from_pretrained(translate_model_id)
|
11 |
-
translator = pipeline(model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='nob_Latn')
|
12 |
|
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,
|
20 |
|
21 |
-
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs=["
|
22 |
|
23 |
if __name__ == "__main__":
|
24 |
iface.launch()
|
|
|
8 |
|
9 |
model = AutoModelForSeq2SeqLM.from_pretrained(translate_model_id)
|
10 |
tokenizer = AutoTokenizer.from_pretrained(translate_model_id)
|
11 |
+
translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='nob_Latn')
|
12 |
|
13 |
def evaluate(input):
|
14 |
eng = caption_eng(input)[0]["generated_text"]
|
15 |
+
nob = translate(eng)[0]["translation_text"]
|
16 |
return (eng, nob)
|
17 |
|
18 |
def translate(text):
|
19 |
+
return translator(text, max_new_tokens=100)
|
20 |
|
21 |
+
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs=[gr.Textbox(label="Caption (English)"), gr.Textbox(label="Caption (Bokmål)")])
|
22 |
|
23 |
if __name__ == "__main__":
|
24 |
iface.launch()
|