Update app.py

app.py

@@ -41,7 +41,7 @@ body, .gradio-container {
     border: 1px solid #333333 !important; 
 }
 
-/* Hide Gradio footer and header folders */
+/* Hide Gradio footer, branding, and header folders */
 footer, header, .gradio-footer, .gradio-header, 
 #footer, #header, #description, 
 .svelte-1i4i8j8, .svelte-1lcdy9p, .svelte-18k01vx, 
@@ -55,6 +55,19 @@ footer, header, .gradio-footer, .gradio-header,
     display: none !important; 
 }
 
+/* Remove image preview icon at the bottom of the input box */
+.gr-image .controls, .gr-image-preview, 
+.gr-image-preview__controls, .gr-image-preview__button { 
+    display: none !important; 
+}
+
+/* Remove the share button in the output box */
+.gradio-container .gr-image .share-btn, 
+.gradio-container .gr-image .share-button, 
+.gradio-container .gr-image .share { 
+    display: none !important; 
+}
+
 /* Make images non-draggable */
 img { 
     pointer-events: none !important;  
@@ -66,7 +79,7 @@ img {
 # ✅ Gradio Interface with Clean UI
 demo = gr.Interface(
     fn=predict,
-    inputs=[gr.Image(type="pil", label="Input")],
+    inputs=[gr.Image(type="pil", label="Input", tool="editor")],  # "tool='editor'" removes preview
     outputs=[gr.Image(type="numpy", label="Restored Image")],
     title="Restore",
     css=custom_css