update

app.py

@@ -61,7 +61,7 @@ def check_input_images(input_images):
         raise gr.Error("No images uploaded!")
 
 @spaces.GPU(duration=120)
-def process_images(input_images, compute_segment_anything):
+def process_images(input_images):
 
     images_path = "/tmp/gradio/images"
     processed_path = "/tmp/gradio/processed"
@@ -81,15 +81,8 @@ def process_images(input_images, compute_segment_anything):
     download_cmd = "tnd-download-data sam --save-dir /tmp/gradio"
     os.system(download_cmd)
 
-    if compute_segment_anything:
-        sam_cmd = " --compute-segment-anything"
-    else:
-        sam_cmd = " --no-compute-segment-anything"
-
     # process the data
     process_data_cmd = f"tnd-process-data initialize --dataset toon3d-dataset --input_path {images_path} --data_prefix {processed_path} --sam_checkpoint_prefix /tmp/gradio/sam-checkpoints"
-    process_data_cmd += sam_cmd
-    print(process_data_cmd)
     os.system(process_data_cmd)
 
     zip_folder = "/tmp/gradio/processed/toon3d-dataset"
@@ -240,9 +233,7 @@ with gr.Blocks(title="Toon3D") as demo:
     gr.Markdown(_HEADER_)
     with gr.Row(variant="panel"):
             input_images = gr.File(label="Upload Images", file_count="multiple", file_types=[".jpg", "jpeg", "png"])
-            with gr.Column():
-                compute_segment_anything = gr.Checkbox(label="Compute Segment Anything? (slow)", value=False)
-                process_data_button = gr.Button("Process Data", elem_id="process_data_button", variant="primary")
+            process_data_button = gr.Button("Process Data", elem_id="process_data_button", variant="primary")
             processed_data_zip = gr.File(label="Processed Data", file_count="single", file_types=[".zip"], interactive=True)
     with gr.Row(variant="panel"):
         labeler_visible = gr.Checkbox(label="Show Labeler", value=False)
@@ -260,7 +251,7 @@ with gr.Blocks(title="Toon3D") as demo:
 
     process_data_button.click(fn=check_input_images, inputs=[input_images]).success(
         fn=process_images,
-        inputs=[input_images, compute_segment_anything],
+        inputs=[input_images],
         outputs=[processed_data_zip],
     )
 
@@ -278,4 +269,4 @@ with gr.Blocks(title="Toon3D") as demo:
 
 if __name__ == "__main__":
     demo.queue(max_size=10)
-    demo.launch()
+    demo.launch(share=True)

requirements.txt

@@ -1,2 +1,2 @@
-toon3d==0.0.3
+toon3d==0.0.4
 segment_anything @ git+https://github.com/facebookresearch/segment-anything.git