Update app.py

app.py

@@ -2,38 +2,75 @@ import os
 import gradio as gr
 from zeroscratches import EraseScratches
 
+# ✅ Install dependencies (if not already installed)
 os.system("pip freeze")
 
-
+# ✅ Inference function
 def predict(img):
+    """ Apply scratch removal on image """
     return EraseScratches().erase(img)
 
+# ✅ Custom CSS for clean design
+custom_css = """
+/* Dark theme styling */
+body, .gradio-container { 
+    background-color: #000000 !important; 
+    color: white !important; 
+}
 
-title = "Zero Scratches"
-description = r"""
-## Old Photo Restoration
-
-This is a lightweight implementation of [Microsoft Bringing Old Photos Back to Life](https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life)
-"""
+/* Styling for buttons */
+.gr-button { 
+    background: #1e90ff !important; 
+    color: white !important; 
+    border: none !important; 
+    padding: 10px 20px !important;   
+    font-size: 14px !important; 
+    width: 100% !important;  
+    cursor: pointer;
+}
 
-article = r"""
-Questions, doubts, comments, please email 📧 `leonelhs@gmail.com`
+/* Hover effect */
+.gr-button:hover {
+    background: #0056b3 !important;
+}
 
-This demo is running on a CPU, if you like this project please make us a donation to run on a GPU or just give us a <a href='https://github.com/leonelhs/zeroscratches/' target='_blank'>Github ⭐</a>
+/* Styling for output boxes */
+.gr-box, .gr-input, .gr-output { 
+    background-color: #1c1c1c !important;  
+    color: white !important; 
+    border: 1px solid #333333 !important; 
+}
 
-<a href="https://www.buymeacoffee.com/leonelhs"><img src="https://img.buymeacoffee.com/button-api/?text=Buy me a coffee&emoji=&slug=leonelhs&button_colour=FFDD00&font_colour=000000&font_family=Cookie&outline_colour=000000&coffee_colour=ffffff" /></a>
+/* Hide Gradio footer and header folders */
+footer, header, .gradio-footer, .gradio-header, 
+#footer, #header, #description, 
+.svelte-1i4i8j8, .svelte-1lcdy9p, .svelte-18k01vx, 
+.svelte-1qtpyo7, .svelte-1xjlq5, .svelte-1g01jrn, 
+.svelte-1rs4uyx, .gradio-branding, 
+#share-btn, #share, #download-btn, #download, 
+.gradio-flag, .gradio-screenshot, 
+.gradio-tab, .gradio-tabs, 
+.gradio-tabs-wrapper, 
+.gradio-header__space { 
+    display: none !important; 
+}
 
-<center><img src='https://visitor-badge.glitch.me/badge?page_id=zeroscratches.visitor-badge' alt='visitor badge'></center>
+/* Make images non-draggable */
+img { 
+    pointer-events: none !important;  
+    -webkit-user-drag: none !important; 
+    user-select: none !important; 
+}
 """
 
+# ✅ Gradio Interface with Clean UI
 demo = gr.Interface(
-    predict, [
-        gr.Image(type="pil", label="Input"),
-    ], [
-        gr.Image(type="numpy", label="Image zero scratches")
-    ],
-    title=title,
-    description=description,
-    article=article)
-
-demo.queue().launch()
+    fn=predict,
+    inputs=[gr.Image(type="pil", label="Input")],
+    outputs=[gr.Image(type="numpy", label="Restored Image")],
+    title="Restore",
+    css=custom_css
+)
+
+# ✅ Launch Gradio app
+demo.queue().launch(debug=True)