Update app.py

app.py

@@ -46,7 +46,7 @@ if uploaded_file:
     # User input for task selection
     task = st.selectbox(
         "Select a task:",
-        ["Generate a caption", "Answer a question", "Detect objects"]
+        ["Generate a caption", "Answer a question", "Detect objects", "Generate segmentation"]
     )
 
     # User prompt
@@ -62,4 +62,4 @@ if uploaded_file:
                 generation = generation[0][input_len:]  # Remove input tokens from output
                 answer = processor.decode(generation, skip_special_tokens=True)
 
-            st.success(f"✅ Result: {answer}")
+            st.success(f"✅ Result: {answer}")