Update app.py

app.py

@@ -3,7 +3,7 @@ from PIL import Image
 import numpy as np
 import torch
 from transformers import AutoImageProcessor, Swin2SRForImageSuperResolution
-import gradio as gr  # Import Gradio for creating the interface
+import gradio as gr
 import spaces
 
 # Function to upscale an image using Swin2SR
@@ -22,7 +22,7 @@ def upscale_image(image, model, processor, device):
         outputs = model(**inputs)
 
     # Move output back to CPU for further processing
-    output = outputs.reconstruction.data.squeeze().cpu().clamp_(0, 1).numpy()
+    output = outputs.reconstruction.data.squeeze().cpu().float().clamp_(0, 1).numpy()
     output = np.moveaxis(output, source=0, destination=-1)
     output_image = (output * 255.0).round().astype(np.uint8)  # Convert from float32 to uint8
 
@@ -66,7 +66,7 @@ interface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        gr.Checkbox(default=True, label="Save as JPEG"),
+        gr.Checkbox(value=True, label="Save as JPEG"),
     ],
     outputs=gr.File(label="Download Upscaled Image"),
     title="Image Upscaler",