Update app.py

app.py

@@ -11,18 +11,24 @@ def generate_3d(image):
         with open(image, "rb") as f:
             response = requests.post(API_URL, headers=headers, files={"file": f})
         if response.status_code == 200:
-            return response.content
+            # Save the model locally
+            output_path = "output.glb"
+            with open(output_path, "wb") as out_file:
+                out_file.write(response.content)
+            return output_path  # Gradio will let user download it
         else:
             return f"Error: {response.text}"
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
+
 iface = gr.Interface(
     fn=generate_3d,
     inputs=gr.Image(type="filepath", label="Upload 2D Image"),
-    outputs="model3d",
+    outputs="file",  # change here
     title="Scan Spectrum - 2D to 3D Converter",
-    description="Upload any 2D image to generate a realistic 3D model you can rotate and zoom.",
+    description="Upload any 2D image to generate a realistic 3D model you can download.",
 )
 
+
 iface.launch()