Update app.py

app.py

@@ -263,6 +263,13 @@ def process_image(input_image):
 def gradio_interface(input_img):
     try:
         depth_raw, normal = process_image(input_img)
+        
+        # Ensure proper format for PNG output
+        if depth_raw is not None:
+            depth_raw = depth_raw.astype(np.uint8)
+        if normal is not None:
+            normal = normal.astype(np.uint8)
+            
         return [depth_raw, normal]
     except Exception as e:
         print(f"Error processing image: {str(e)}")
@@ -273,8 +280,8 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.Image(label="Input Image"),
     outputs=[
-        gr.Image(label="Raw Depth Map"),
-        gr.Image(label="DirectX Normal Map")
+        gr.Image(label="Raw Depth Map", type="numpy", format="png"),
+        gr.Image(label="DirectX Normal Map", type="numpy", format="png")
     ],
     title="Depth and Normal Map Generation",
     description="Upload an image to generate its depth map and normal map.",