Update app.py

app.py

@@ -44,6 +44,7 @@ def get_transform(resolution):
     ])
 
 # Image-to-Text Function
+@spaces.GPU
 def image_to_text(image, prompt, resolution=1024, steps=64, cfg=9.0):
     try:
         transform = get_transform(resolution)
@@ -73,6 +74,7 @@ def image_to_text(image, prompt, resolution=1024, steps=64, cfg=9.0):
         return f"Error: {str(e)}"
 
 # Text-to-Image Function
+@spaces.GPU
 def text_to_image(prompt, negative_prompt, num_images=1, resolution=1024, steps=64, cfg=9.0):
     try:
         negative_prompt = negative_prompt or "worst quality, low quality, low res, blurry, distortion, watermark, logo, signature, text, jpeg artifacts, signature, sketch, duplicate, ugly, identifying mark"