Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 #from tempfile import NamedTemporaryFile
 import numpy as np
+import spaces
 import random
 import string
 from diffusers import StableDiffusionPipeline as DiffusionPipeline
@@ -60,6 +61,7 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
+@spaces.GPU
 def infer(prompt):
     name = generate_random_string(12)+".png"
     english_prompt = translate(prompt,"en")