Update app.py

app.py

@@ -3,18 +3,17 @@ import gradio as gr
 import torch
 from transformers import pipeline
 from PIL import Image
-import traceback
 
 # Global model storage
 models = {}
 
 @spaces.GPU(duration=120)
 def generate_code(image, model_choice, prompt_style):
-    """Generate CADQuery code - single GPU function."""
+    """Generate CADQuery code from image."""
     if image is None:
         return "❌ Please upload an image first."
     
-    # Create prompts
+    # Prompts
     prompts = {
         "Simple": "Generate CADQuery Python code for this 3D model:",
         "Detailed": "Analyze this 3D CAD model and generate Python CADQuery code.\n\nRequirements:\n- Import cadquery as cq\n- Store result in 'result' variable\n- Use proper CADQuery syntax\n\nCode:",
@@ -93,7 +92,7 @@ python your_script.py
         return f"❌ **Generation Failed**: {str(e)[:500]}"
 
 def system_info():
-    """Get system info - no GPU needed."""
+    """Get system info."""
     info = f"""## 🖥️ System Information
 
 - **CUDA Available**: {torch.cuda.is_available()}
@@ -148,5 +147,4 @@ with gr.Blocks(title="GLM-4.5V CAD Generator", theme=gr.themes.Soft()) as demo:
 if __name__ == "__main__":
     print("🚀 Starting GLM-4.5V CAD Generator...")
     print(f"CUDA available: {torch.cuda.is_available()}")
-    # Remove share=True for HF Spaces
-    demo.launch()
+    demo.launch()  # NO share=True!