Update app.py

app.py

@@ -1,4 +1,4 @@
-import import gradio as gr
+import gradio as gr
 import replicate
 import requests
 import os
@@ -11,11 +11,13 @@ def generate_3d(prompt):
     try:
         output = replicate.run(
             "openai/shap-e:9c9ecb8d35d0cc0ecde68b8a0d4a2f94f2bc6ee9393e8c3d76c44d7c0d75b1f9",
-            input={"prompt": prompt, "guidance_scale": 15, "num_inference_steps": 64}
+            input={
+                "prompt": prompt,
+                "guidance_scale": 15,
+                "num_inference_steps": 64
+            }
         )
 
-        print("REPLICATE OUTPUT:", output)
-
         if not output or not isinstance(output, list):
             return "Error: No model output."
 
@@ -23,7 +25,7 @@ def generate_3d(prompt):
         response = requests.get(glb_url)
 
         if response.status_code != 200:
-            return f"Failed to download GLB file. Status code: {response.status_code}"
+            return f"Download failed: {response.status_code}"
 
         file_name = f"{uuid.uuid4()}.glb"
         with open(file_name, 'wb') as f: