Update app.py

app.py

@@ -39,6 +39,7 @@ from safetensors.torch import load_file
 import torch
 #i#mport torch.distributed as dist
 from torchvision.transforms.functional import to_pil_image
+import spaces
 
 from imgproc import generate_crop_size_list
 import models
@@ -365,7 +366,7 @@ def find_free_port() -> int:
     sock.close()
     return port
 
-@spaces.GPU(duration=60)
+
 def main():
     parser = argparse.ArgumentParser()
 
@@ -549,6 +550,7 @@ def main():
                 label="Examples",
             )  # noqa
 
+        @spaces.GPU(duration=200)
         def on_submit(*args):
             for q in request_queues:
                 q.put(args)