update description

app.py

@@ -1,5 +1,5 @@
 # --- Environment Variables Used ---
-# ENABLE_ZEROGPU: Set to 'true' or '1' to enable @spaces.GPU decorator (for Hugging Face Spaces).
+# DISABLE_ZEROGPU: Set to 'true' or '1' to disable @spaces.GPU decorator (for Hugging Face Spaces).
 # TRIPOSG_CODE_PATH: Absolute path to a local directory containing the checked-out TripoSG repository (scribble branch).
 # GITHUB_TOKEN: A GitHub token used for cloning the TripoSG repo if TRIPOSG_CODE_PATH is not provided.
 # WEIGHTS_PATH: Absolute path to a local directory containing the TripoSG-scribble model weights.
@@ -257,7 +257,7 @@ class WaifuDiffusionInterrogator:
         new_image = Image.new('RGBA', image.size, 'WHITE')
         new_image.paste(image, mask=image)
         image = new_image.convert('RGB')
-        image.save("image_to_wd.png")
+
         image = np.asarray(image)
         image = image[:, :, ::-1]
 
@@ -327,13 +327,13 @@ def generate_3d(scribble_image_dict, prompt, scribble_confidence, text_confidenc
                     input_prompt = generated_prompt
                 else:
                     print("WD14 tagging did not produce tags above threshold.")
-                    input_prompt = "object" # Fallback prompt
+                    input_prompt = "3d object" # Fallback prompt
             else:
                 print("WD14 tagging failed or tagger not initialized.")
-                input_prompt = "object" # Fallback prompt
+                input_prompt = "3d object" # Fallback prompt
         except Exception as e:
             print(f"Error during WD14 tagging: {e}")
-            input_prompt = "object" # Fallback prompt
+            input_prompt = "3d object" # Fallback prompt
     else:
         print(f"Using user provided prompt: {input_prompt}")
     # --- End Prompt Handling ---
@@ -398,20 +398,27 @@ def generate_3d(scribble_image_dict, prompt, scribble_confidence, text_confidenc
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# TripoSG Scribble!!")
-    gr.Markdown("3D model generation with simple scribble and text prompt.") # Updated guidance
+    gr.Markdown("""
+[![GitHub](https://img.shields.io/badge/GitHub-%23121011.svg?logo=github&logoColor=white)](https://github.com/VAST-AI-Research/TripoSG)
+
+### Fast 3D shape prototyping with simple scribble and text prompt. Presented by [Tripo](https://www.tripo3d.ai/).
+
+- For local deployment, simply clone this space, set up the environment and run with DISABLE_ZEROGPU=1.
+- Feel free to tune the scribble confidence to balance fidelity and alignment :)
+""")
     with gr.Row():
         with gr.Column(scale=1):
             image_input = gr.ImageEditor(
                 label="Scribble Input (Draw Black on White)",
                 value=initial_value,
                 image_mode="RGB",
-                brush=gr.Brush(default_color="#000000", color_mode="fixed", default_size=5), # Fixed small brush size
+                brush=gr.Brush(default_color="#000000", color_mode="fixed", default_size=5),
                 interactive=True,
-                eraser=gr.Brush(default_color="#FFFFFF", color_mode="fixed", default_size=20) # Fixed small eraser size
+                eraser=gr.Brush(default_color="#FFFFFF", color_mode="fixed", default_size=20)
             )
             prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., a cute cat wearing a hat")
             confidence_input = gr.Slider(minimum=0.0, maximum=1.0, value=0.4, step=0.05, label="Scribble Confidence")
-            prompt_confidence_input = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, step=0.05, label="Prompt Confidence") # Added slider
+            prompt_confidence_input = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, step=0.05, label="Prompt Confidence")
             seed_input = gr.Number(label="Seed", value=0, precision=0) # Added Seed input back
             with gr.Row():
                 submit_button = gr.Button("Generate 3D Model", variant="primary", scale=1)