Update app.py

app.py

@@ -1,24 +1,15 @@
+import spaces
 import gradio as gr
 import cv2
 import numpy
 import os
 import random
-import torch
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from basicsr.utils.download_util import load_file_from_url
+
 from realesrgan import RealESRGANer
 from realesrgan.archs.srvgg_arch import SRVGGNetCompact
-from torchvision.transforms.functional import rgb_to_grayscale
-import spaces
 
-# Enable HF Spaces GPU wrapper only when explicitly requested
-# Set env USE_SPACES_GPU=1 on Hugging Face Spaces to enable GPU decorator
-def maybe_spaces_gpu(fn):
-    use_gpu_decorator = os.getenv("USE_SPACES_GPU", "0") == "1"
-    if use_gpu_decorator:
-        import spaces
-        return spaces.GPU(duration=120)(fn)
-    return fn
 
 last_file = None
 img_mode = "RGBA"
@@ -72,17 +63,16 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
         dni_weight = [denoise_strength, 1 - denoise_strength]
 
     # Restorer Class
-    use_cuda = torch.cuda.is_available()
     upsampler = RealESRGANer(
         scale=netscale,
         model_path=model_path,
         dni_weight=dni_weight,
         model=model,
-        tile=128,
+        tile=0,
         tile_pad=10,
         pre_pad=10,
-        half=use_cuda,
-        gpu_id=0 if use_cuda else None
+        half=False,
+        gpu_id=None
     )
 
     # Use GFPGAN for face enhancement
@@ -109,10 +99,14 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
         print('Error', error)
         print('If you encounter CUDA out of memory, try to set --tile with a smaller number.')
     else:
-        extension = 'jpg'
+        # Save restored image and return it to the output Image component
+        if img_mode == 'RGBA':  # RGBA images should be saved in png format
+            extension = 'png'
+        else:
+            extension = 'jpg'
 
         out_filename = f"output_{rnd_string(8)}.{extension}"
-        cv2.imwrite(out_filename, output, [int(cv2.IMWRITE_JPEG_QUALITY), 92])
+        cv2.imwrite(out_filename, output)
         global last_file
         last_file = out_filename
         return out_filename
@@ -178,14 +172,10 @@ def image_properties(img):
 
 def main():
     # Gradio Interface
-    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme=gr.themes.Soft()) as demo:
+    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme="ParityError/Interstellar") as demo:
 
         gr.Markdown(
-            """# <div align="center"> Ilaria Upscaler 💖 </div>  
-
-        Do not use images over 750x750 especially with 4x the resolution upscaling, it will give you an error.
-        
-        Hugginface port of [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN).
+            """ Image Upscaler
         """
         )
 
@@ -221,12 +211,11 @@ def main():
         # Undocumented method to clear a component's value using Javascript
 
         gr.Markdown(
-            """Made with love by Ilaria 💖 | Support me on [Ko-Fi](https://ko-fi.com/ilariaowo) | Join [AI Hub](https://discord.gg/aihub)
+            """
         """
         )
 
-    demo.queue(max_size=10)
-    demo.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False, share=False, show_error=True)
+    demo.launch()
 
 
 if __name__ == "__main__":