Update app.py

app.py

@@ -32,7 +32,7 @@ def caption_and_translate(img, min_len, max_len):
     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=min_len, max_length=max_len)
+    out_blip = model_blip.generate(**inputs_blip, min_length=70, max_length=1000)
     english_caption = processor_blip.decode(out_blip[0], skip_special_tokens=True)
 
     # Translate caption from English to Arabic
@@ -47,9 +47,9 @@ def caption_and_translate(img, min_len, max_len):
 # Gradio interface with multiple outputs
 img_cap_en_ar = gr.Interface(
     fn=caption_and_translate,
-    inputs=[gr.Image(type='filepath', label='Image'),
-            gr.Slider(label='Minimum Length', minimum=1, maximum=500, value=30),
-            gr.Slider(label='Maximum Length', minimum=1, maximum=500, value=100)],
+    inputs=[gr.Image(type='filepath', label='Image')]
+            #gr.Slider(label='Minimum Length', minimum=1, maximum=500, value=30),
+            #gr.Slider(label='Maximum Length', minimum=1, maximum=500, value=100)],
     outputs=[gr.Textbox(label='English Caption'),
              gr.HTML(label='Arabic Caption')],
     title='Image Captioning | وصف الصورة',
@@ -88,7 +88,7 @@ text_recognition = gr.Interface(
     outputs=[gr.Textbox(label='Extracted text'), gr.HTML(label= 'Translateted of Extracted text ')],   # Output is text
     title="Text Extraction and Translation | إستخراج النص وترجمتة",
     description="Upload an image then Submet to extract text and translate it to Arabic| قم برفع الصورة وأرسلها ليظهر لك النص من الصورة",
-    examples =[["tx_image_0.png"]],
+    examples =[["image_0.png"], ["image_1.png"]]
 )
 
 # Load trocr model for handwritten text extraction
@@ -120,7 +120,7 @@ handwritten_rec = gr.Interface(
              gr.HTML(label='Arabic Text')],
     title="Handwritten Text Extraction | | إستخراج النص المكتوب بخط اليد وترجمتة",
     description="Upload an image then Submet to extract text and translate it to Arabic| قم برفع الصورة وأرسلها ليظهر لك النص من الصورة",
-    examples =[["tx_image_1.png"]]
+    examples =[["tx_image_1.png"], ["tx_image_3.png"]]
 )
 
 # Combine all interfaces into a tabbed interface