Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from gradio_pannellum import Pannellum
 import torch
@@ -16,11 +17,13 @@ img2panoimg = Image2360PanoramaImagePipeline(model_path, torch_dtype=torch.float
 # Load the default mask image
 default_mask = Image.open("i2p-mask.jpg").convert("RGB")
 
+@spaces.GPU(duration=200)
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale}
     output = txt2panoimg(input_data)
     return output
 
+@spaces.GPU(duration=200)
 def image_to_pano(image, mask, prompt, upscale):
     image = image.resize((512, 512))
     if mask is None: