Update app.py

app.py

@@ -65,14 +65,15 @@ with gr.Blocks(css="style.css") as demo:
     )
     seg_inputs = [
         image,
-        seg_input
+        seg_input,
+        max_new_tokens
         ]
     seg_outputs = [
         annotated_image
     ]
     seg_btn.click(
         fn=parse_segmentation,
-        inputs=[seg_inputs, "max_new_tokens"],
+        inputs=seg_inputs,
         outputs=seg_outputs,
     )
   with gr.Tab("Text Generation"):