Update app.py

app.py

@@ -31,7 +31,16 @@ def caption_and_translate(img, min_len, max_len):
     # Generate English caption
     raw_image = Image.open(img).convert('RGB')
     inputs_blip = processor_blip(raw_image, return_tensors="pt")
-    out_blip = model_blip.generate(**inputs_blip, min_length=5, max_length=50)
+    # Create GenerationConfig with max_length
+    generation_config = GenerationConfig(max_length=max_len)
+
+    out_blip = model_blip.generate(
+        **inputs_blip,
+        min_length=min_len,
+        max_length=max_len,
+        config=generation_config  # Pass the generation_config object here
+    )
+    #out_blip = model_blip.generate(**inputs_blip, min_length=5, max_length=50)
     english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
 
     # Translate caption from English to Arabic