Update app.py
Browse files
app.py
CHANGED
@@ -34,14 +34,11 @@ def caption_and_translate(img, min_len, max_len):
|
|
34 |
out_blip = model_blip.generate(**inputs_blip, min_length=min_len, max_length=max_len)
|
35 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
36 |
|
37 |
-
|
38 |
# Translate caption from English to Arabic
|
39 |
arabic_caption = translate(english_caption)
|
40 |
arabic_caption = arabic_caption[0]['translation_text']
|
41 |
-
|
42 |
translated_caption = f'<div dir="rtl">{arabic_caption}</div>'
|
43 |
|
44 |
-
|
45 |
# Return both captions
|
46 |
return english_caption, translated_caption
|
47 |
|
|
|
34 |
out_blip = model_blip.generate(**inputs_blip, min_length=min_len, max_length=max_len)
|
35 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
36 |
|
|
|
37 |
# Translate caption from English to Arabic
|
38 |
arabic_caption = translate(english_caption)
|
39 |
arabic_caption = arabic_caption[0]['translation_text']
|
|
|
40 |
translated_caption = f'<div dir="rtl">{arabic_caption}</div>'
|
41 |
|
|
|
42 |
# Return both captions
|
43 |
return english_caption, translated_caption
|
44 |
|