Update app.py

app.py

@@ -1,5 +1,11 @@
 # Code taken and  adopted from https://huggingface.co/spaces/havas79/Real-ESRGAN_Demo/blob/main/app.py - credit where credit is due.
 
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 import gradio as gr
 import cv2
 import numpy
@@ -62,7 +68,7 @@ def upscale(img):
         tile_pad=10,
         pre_pad=10,
         half=False,
-        gpu_id=None,
+        gpu_id=0,
     )
 
     # Convert the input PIL image to cv2 image, so that it can be processed by realesrgan