gpu duration update

app.py

@@ -45,7 +45,7 @@ aihub_deplot_model_path='./deplot_k.pt'
 t5_model_path = './ke_t5.pt'
 
 # Load first model ko-deplot
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def load_model1():
     processor1 = Pix2StructProcessor.from_pretrained('nuua/ko-deplot')
     model1 = Pix2StructForConditionalGeneration.from_pretrained('nuua/ko-deplot')
@@ -56,7 +56,7 @@ def load_model1():
 processor1,model1=load_model1()
 
 # Load second model aihub-deplot
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def load_model2():
     processor2 = AutoProcessor.from_pretrained("ybelkada/pix2struct-base")
     model2 = Pix2StructForConditionalGeneration.from_pretrained("ybelkada/pix2struct-base")
@@ -68,7 +68,7 @@ processor2,model2=load_model2()
 
 
 #Load third model unichart
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def load_model3():
     unichart_model_path = "./unichart"
     model3 =  VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
@@ -84,7 +84,7 @@ def format_output(prediction):
     return prediction.replace('<0x0A>', '\n')
 
 # First model prediction ko-deplot
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def predict_model1(image):
     images = [image]
     inputs = processor1(images=images, text="What is the title of the chart", return_tensors="pt", padding=True)
@@ -117,7 +117,7 @@ def replace_unk(text):
     return text
 
 # Second model prediction aihub_deplot
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def predict_model2(image):
     image = image.convert("RGB")
     inputs = processor2(images=image, return_tensors="pt", max_patches=MAX_PATCHES).to(device)
@@ -134,7 +134,7 @@ def predict_model2(image):
     refined_table = replace_unk(generated_datatable)
     return refined_table
 
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True,duration=120)
 def predict_model3(image):
     image=image.convert("RGB")
     input_prompt = "<extract_data_table> <s_answer>"