Update app.py

app.py

@@ -36,7 +36,12 @@ def process_image(image, num_beams=5, min_p=0.0, top_p=1.0):
             image = image.convert("RGB")
 
         # Prepare inputs for the model
-        inputs = processor(text="<CAPTION>", images=image, return_tensors="pt")
+        inputs = processor(
+            text="<CAPTION>",
+            images=image,
+            return_tensors="pt"
+        )
+
         # Move tensors to the appropriate device
         inputs = {k: v.to(device) for k, v in inputs.items()}
 
@@ -74,19 +79,32 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(label="Input Picture")
+
             with gr.Column():
                 output_text = gr.Textbox(label="Output Text")
 
                 submit_btn = gr.Button(value="Submit")
 
                 num_beams_slider = gr.Slider(
-                    minimum=1, maximum=5, step=1, value=5, label="Number of Beams"
+                    minimum=1,
+                    maximum=5,
+                    step=1,
+                    value=5,
+                    label="Number of Beams"
                 )
                 min_p_slider = gr.Slider(
-                    minimum=0, maximum=1, step=0.01, value=0.0, label="Min-P"
+                    minimum=0,
+                    maximum=1,
+                    step=0.01,
+                    value=0.0,
+                    label="Min-P"
                 )
                 top_p_slider = gr.Slider(
-                    minimum=0, maximum=1, step=0.01, value=1.0, label="Top-P"
+                    minimum=0,
+                    maximum=1,
+                    step=0.01,
+                    value=1.0,
+                    label="Top-P"
                 )
 
         gr.Examples(