Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 from PIL import Image
 from RealESRGAN import RealESRGAN
 import gradio as gr
+import spaces
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model2 = RealESRGAN(device, scale=2)
@@ -12,6 +13,7 @@ model8 = RealESRGAN(device, scale=8)
 model8.load_weights('weights/RealESRGAN_x8.pth', download=True)
 
 
+@spaces.GPU
 def inference(image, size):
     global model2
     global model4