Update app.py

app.py

@@ -8,7 +8,10 @@ from PIL import Image
 from realesrgan import RealESRGANer
 from huggingface_hub import hf_hub_download
 import gradio as gr
+import torch
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+half = True if device == "cuda" else False
 model = RRDBNet(num_in_ch=3, num_out_ch=3)
 upsampler = RealESRGANer(
     scale=4,
@@ -16,7 +19,7 @@ upsampler = RealESRGANer(
     model=model,
     tile=0,
     pre_pad=0,
-    half=True,
+    half=half,
 )
 
 def upsample(image):