update

app.py

@@ -11,11 +11,11 @@ import threading
 viewer_thread_instance = None
 stop_event = threading.Event()
 shared_url = None
-compute_segment_anything = False
+compute_segment_anything_value = False
 
 _HEADER_ = '''
 <h2>Toon3D: Seeing Cartoons from a New Perspective</h2>
-Toon3D lifts cartoons into 3D via aligning and warping backprojected monocular depth predictions. The project page is at <a href='https://toon3d.studio/' target='_blank'>https://toon3d.studio/</a> and the Toon3D Labeler is at <a href='https://labeler.toon3d.studio/' target='_blank'>https://labeler.toon3d.studio/</a>. Follow the steps below to run Toon3D!
+Toon3D lifts hand-drawn images to 3D with a piecewise-rigid deformation optimization at hand-labeled keypoints and using monocular depth as a prior. The project page is at <a href='https://toon3d.studio/' target='_blank'>https://toon3d.studio/</a> and the Toon3D Labeler is at <a href='https://labeler.toon3d.studio/' target='_blank'>https://labeler.toon3d.studio/</a>. Follow the steps below to run Toon3D!
 
 <div style="margin-top: 20px; font-size: 16px; line-height: 1.6;">
     <div style="display: flex; justify-content: space-between;">
@@ -62,7 +62,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, compute_segment_anything_value):
 
     images_path = "/tmp/gradio/images"
     processed_path = "/tmp/gradio/processed"
@@ -82,10 +82,10 @@ 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:
+    if compute_segment_anything_value:
         sam_cmd = " --compute-segment-anything"
     else:
-        sam_cmd = ""
+        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"
@@ -105,8 +105,8 @@ def toggle_labeler_visibility(visible):
         return ""
     
 def set_compute_segment_anything(value):
-    global compute_segment_anything
-    compute_segment_anything = value
+    global compute_segment_anything_value
+    compute_segment_anything_value = value
 
 def check_input_toon3d(processed_data_zip, labeled_data):
     if processed_data_zip is None:
@@ -266,7 +266,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, compute_segment_anything_value],
         outputs=[processed_data_zip],
     )