Update app.py

app.py

@@ -6,11 +6,11 @@ from PIL import Image
 import numpy as np
 import os
 
-# ✅ Check if CUDA is available
+# Check if CUDA is available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 print(f"Using device: {device}")
 
-# ✅ Model definition
+# Model definition
 class DeblurNet(nn.Module):
     def __init__(self):
         super(DeblurNet, self).__init__()
@@ -53,7 +53,7 @@ class DeblurNet(nn.Module):
         x = self.final_conv(x)
         return torch.tanh(x)
 
-# ✅ Load model
+# Load model
 model = DeblurNet().to(device)
 model_path = os.path.join('model', 'best_deblur_model.pth')
 
@@ -65,7 +65,7 @@ if os.path.exists(model_path):
 else:
     print(f"Model file not found at {model_path}. Please check the path.")
 
-# ✅ Image processing functions
+# Image processing functions
 transform = transforms.Compose([
     transforms.Resize((256, 256)),
     transforms.ToTensor(),
@@ -111,18 +111,10 @@ def deblur_image(filepath):
         print(f"Error processing image: {e}")
         return None
 
-# ✅ Custom CSS for styling, hiding share & fullscreen icons
+# Custom CSS for styling and hiding fullscreen and share buttons
 custom_css = """
-/* Hide Gradio's footer and header */
-footer, header, .gradio-footer, .gradio-header {
-    display: none !important;
-}
-/* Hide share button */
-.share-wrap, .gr-share-btn {
-    display: none !important;
-}
-/* Hide fullscreen icon */
-.gr-button[title="Fullscreen"] {
+/* Hide fullscreen and share buttons in the output box */
+.output_image .fullscreen, .output_image .share {
     display: none !important;
 }
 /* Non-draggable images */
@@ -154,7 +146,7 @@ body, .gradio-container {
 }
 """
 
-# ✅ Gradio interface
+# Gradio interface
 demo = gr.Interface(
     fn=deblur_image,
     inputs=gr.File(label="Input", type="filepath"),
@@ -164,6 +156,6 @@ demo = gr.Interface(
     css=custom_css
 )
 
-# ✅ Launch Gradio app
+# Launch Gradio app
 if __name__ == "__main__":
     demo.launch()