interface

app.py

@@ -37,8 +37,16 @@ def segment_image(img: PIL.Image.Image, classnames: str, use_lposs_plus: bool |
 
 demo = gr.Interface(
     fn=segment_image,
-    inputs=["image", "text", "checkbox"],
-    outputs=["annotatedimage"],
+    inputs=[
+        gr.Image(label="Input Image"),
+        gr.Textbox(label="Class Names", info="Separate class names with commas"),
+        gr.Checkbox(label="Use LPOSS+", info="Enable pixel-level refinement using LPOSS+")
+    ],
+    outputs=[
+        gr.AnnotatedImage(label="Segmentation Results")
+    ],
+    title="LPOSS: Label Propagation Over Patches and Pixels for Open-vocabulary Semantic Segmentation",
+    description="Upload an image and specify the objects you want to segment by listing their names separated by commas.",
 )
 
 demo.launch()