Update app.py

app.py

@@ -1,10 +1,13 @@
+import os
+import warnings
+from pathlib import Path
+
 import gradio as gr
 import torch
 import torch.nn as nn
 import torchvision.transforms as transforms
 from PIL import Image
 import numpy as np
-import os
 
 # Check if CUDA is available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -111,10 +114,10 @@ def deblur_image(filepath):
         print(f"Error processing image: {e}")
         return None
 
-# Custom CSS to hide fullscreen and share icons
+# ✅ Custom CSS to hide fullscreen and share icons
 custom_css = """
-/* Hide fullscreen and share icons in the output image component */
-.gr-button[title="Fullscreen"], .gr-button[title="Share"] {
+/* Hide fullscreen and share buttons */
+button[title="Fullscreen"], button[title="Share"] {
     display: none !important;
 }
 /* Hide Gradio's footer and header */
@@ -150,7 +153,7 @@ body, .gradio-container {
 }
 """
 
-# Gradio interface
+# ✅ Gradio interface
 demo = gr.Interface(
     fn=deblur_image,
     inputs=gr.File(label="Input", type="filepath"),
@@ -160,6 +163,6 @@ demo = gr.Interface(
     css=custom_css
 )
 
-# Launch Gradio app
+# ✅ Launch Gradio app
 if __name__ == "__main__":
     demo.launch()