Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import torch
 from PIL import Image
 import random
+import spaces
 
 import requests
 
@@ -142,7 +143,8 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui,disable_au
             neg_prompt=f"(((deformed))), {neg_prompt}, girl, boy, photo, people, low quality, ui, error, lowres, jpeg artifacts, 2d, 3d, cg, text"
 
     return prompt,neg_prompt
-    
+
+@spaces.GPU
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator,superreso=False):
     global pipe, upscaler
     if(superreso):
@@ -179,6 +181,7 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator,sup
     
     return result.images[0]
 
+@spaces.GPU
 def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator,superreso=False):
     ratio = min(height / img.height, width / img.width)
     img = img.resize((int(img.width * ratio), int(img.height * ratio)), Image.LANCZOS)