Update app.py

app.py

@@ -38,10 +38,13 @@ def get_results(image, prompt,segment):
 
     # Prepare the payload (Adjust this part according to the API requirements)
     #payload = json.dumps({"base64str": base64str, "classes": prompt})
+    task_="0"
+    if segment == "Segmentation":
+        task_="1"
     payload = {
         "base64str": base64str,
         "classes": prompt,
-        "segment": "1" if segment else "0"  # 1 if checkbox is selected, 0 if not
+        "segment": task_  # 1 if checkbox is selected, 0 if not
     }
     # Send the request to the API
     response = requests.put(api_key, data=payload)
@@ -142,7 +145,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         text_input = gr.inputs.Textbox(label="Prompt")
     with gr.Row():
-        segment_checkbox = gr.inputs.Checkbox(label="Segment", default=False)
+        segment_checkbox = gr.Radio(["Segmentation", "Detection"], value="file",
+                                         label="Condition on a task (optional) Segmentation or Detection")
+        #segment_checkbox = gr.inputs.Checkbox(label="Segment", default=False)
     with gr.Row():
         button = gr.Button("Run")