Update app.py

app.py

@@ -43,3 +43,16 @@ def inference(input_image: Image.Image) -> Image.Image:
     except Exception as e:
         print(f"Error during image enhancement: {e}")
         return input_image  # Return the original image if there's an error
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=inference,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="Sketch to Floor Plan Pix2Pix",
+    description="Upload a sketch image and the model will generate a floor plan."
+)
+
+# Launch the Gradio interface when the script is run directly
+if __name__ == "__main__":
+    iface.launch()