Kawthar12h
commited on
Commit
•
b8382c9
1
Parent(s):
d2b6347
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def caption_and_translate(img, min_len, max_len):
|
|
31 |
# Generate English caption
|
32 |
raw_image = Image.open(img).convert('RGB')
|
33 |
inputs_blip = processor_blip(raw_image, return_tensors="pt")
|
34 |
-
out_blip = model_blip.generate(**inputs_blip, min_length=
|
35 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
36 |
|
37 |
# Translate caption from English to Arabic
|
|
|
31 |
# Generate English caption
|
32 |
raw_image = Image.open(img).convert('RGB')
|
33 |
inputs_blip = processor_blip(raw_image, return_tensors="pt")
|
34 |
+
out_blip = model_blip.generate(**inputs_blip, min_length=5, max_length=50)
|
35 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
36 |
|
37 |
# Translate caption from English to Arabic
|