- Added cpu and Cuda usage

app.py

@@ -3,6 +3,8 @@ import spaces
 from gradio_imageslider import ImageSlider
 from image_gen_aux import UpscaleWithModel
 from image_gen_aux.utils import load_image
+import torch
+
 
 # This uses https://github.com/asomoza/image_gen_aux/blob/main/src/image_gen_aux/upscalers/README.md
 # Also this space has been duplicated from their official huggingface space, https://huggingface.co/spaces/OzzyGT/basic_upscaler
@@ -118,7 +120,9 @@ MODELS = {
 def upscale_image(image, model_selection):
     original = load_image(image)
 
-    upscaler = UpscaleWithModel.from_pretrained(MODELS[model_selection]).to("cuda")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    upscaler = UpscaleWithModel.from_pretrained(MODELS[model_selection]).to(device)
+
     image = upscaler(original, tiling=True, tile_width=1024, tile_height=1024)
 
     return original, image
@@ -165,4 +169,4 @@ with gr.Blocks() as demo:
     )
 
 
-demo.launch(share=False)
+demo.launch(share=True)