Update app.py

app.py

@@ -268,80 +268,29 @@ def gradio_interface(input_img):
         print(f"Error processing image: {str(e)}")
         return [None, None]
 
-# Define default values
-DEFAULT_VALUES = {
-    "prompt": "clean, texture, high-resolution, 8k",
-    "negative_prompt": "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
-    "guidance_scale": 2.5,
-    "conditioning_scale": 1.0,
-    "num_steps": 6,
-    "seed": None,
-    "upscale_factor": 4,
-    "color_fix_method": "adain"
-}
-
-# Create Gradio interface
-demo = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.Image(label="Input Image"),
-        gr.Textbox(
-            label="Prompt", 
-            value=DEFAULT_VALUES["prompt"],
-            default=DEFAULT_VALUES["prompt"]
-        ),
-        gr.Textbox(
-            label="Negative Prompt", 
-            value=DEFAULT_VALUES["negative_prompt"],
-            default=DEFAULT_VALUES["negative_prompt"]
-        ),
-        gr.Slider(
-            minimum=1.0, 
-            maximum=20.0, 
-            value=DEFAULT_VALUES["guidance_scale"],
-            default=DEFAULT_VALUES["guidance_scale"],
-            label="Guidance Scale"
-        ),
-        gr.Slider(
-            minimum=0.1, 
-            maximum=2.0, 
-            value=DEFAULT_VALUES["conditioning_scale"],
-            default=DEFAULT_VALUES["conditioning_scale"],
-            label="Conditioning Scale"
-        ),
-        gr.Slider(
-            minimum=1, 
-            maximum=50, 
-            value=DEFAULT_VALUES["num_steps"],
-            default=DEFAULT_VALUES["num_steps"],
-            step=1, 
-            label="Number of Steps"
-        ),
-        gr.Number(
-            label="Seed", 
-            value=DEFAULT_VALUES["seed"],
-            default=DEFAULT_VALUES["seed"]
-        ),
-        gr.Slider(
-            minimum=1, 
-            maximum=4, 
-            value=DEFAULT_VALUES["upscale_factor"],
-            default=DEFAULT_VALUES["upscale_factor"],
-            step=1, 
-            label="Upscale Factor"
-        ),
-        gr.Radio(
-            choices=["none", "wavelet", "adain"], 
-            label="Color Fix Method", 
-            value=DEFAULT_VALUES["color_fix_method"],
-            default=DEFAULT_VALUES["color_fix_method"]
-        ),
+# Define interface
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=gr.Image(label="Input Image"),
+    outputs=[
+        gr.Image(label="Raw Depth Map"),
+        gr.Image(label="Normal Map")
     ],
-    outputs=gr.Image(label="Generated Image"),
-    title="Controllable Conditional Super-Resolution",
-    description="Upload an image to enhance its resolution using CCSR."
+    title="Depth and Normal Map Generation",
+    description="Upload an image to generate its depth map and normal map.",
+    examples=[
+        "PavingStones144_1K-PNG_Color.png",
+        "0269B55506557D8D_diffuse.png",
+        "Brick_Floor_tkqlcekcw_8K_surface_msAlbedo_baked.jpg",
+        "Brick_Painted_sb0hkjp0_4K_surface_msAlbedo_baked.jpg",
+        "Concrete_rlvlbep0_4K_surface_msAlbedo_baked.jpg",
+        "Grass_Dried_scmkvwp0_4K_surface_msAlbedo_baked.jpg",
+        "Stone_Tile_uc2jdbpg_8K_surface_msAlbedo_baked.jpg",
+        "Surface_Tiles_smgmjog_8K_surface_msAlbedo_baked.jpg",
+        "Panel.jpg"
+    ]
 )
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()