Update app.py

app.py

@@ -98,7 +98,6 @@ def img2b64(image_path):
     data_uri = f"data:image/jpeg;base64,{b64}"
     return data_uri
 
-@spaces.GPU
 def initialize_models(args):
     os.makedirs("tmp", exist_ok=True)
     # Paths
@@ -209,7 +208,7 @@ def process_large_image(raw_img):
     else:
         return raw_img
 
-@spaces.GPU
+@spaces.GPU(duration=200)
 def chat_step(image1, image2, text, height, width, steps, guidance,
               ocr_enhancer, joint_with_t5, enhance_generation, enhance_understanding,
               seed, num_imgs, history_state, progress=gr.Progress()):