Kawthar12h commited on
Commit
b8382c9
1 Parent(s): d2b6347

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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
 
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