Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,16 +31,8 @@ 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 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
out_blip = model_blip.generate(
|
| 38 |
-
**inputs_blip,
|
| 39 |
-
min_length=min_len,
|
| 40 |
-
max_length=max_len,
|
| 41 |
-
config=generation_config # Pass the generation_config object here
|
| 42 |
-
)
|
| 43 |
-
#out_blip = model_blip.generate(**inputs_blip, min_length=5, max_length=50)
|
| 44 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
| 45 |
|
| 46 |
# 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 |
+
|
| 35 |
+
out_blip = model_blip.generate(**inputs_blip, min_length=50, max_length=500)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
|
| 37 |
|
| 38 |
# Translate caption from English to Arabic
|