Update app.py

app.py

@@ -21,8 +21,6 @@ import logging
 import subprocess
 import spaces
 
-@spaces.GPU
-
 # Git LFS pull 명령어 실행
 result = subprocess.run(['git', 'lfs', 'pull'], capture_output=True, text=True)
 
@@ -78,6 +76,7 @@ def format_output(prediction):
     return prediction.replace('<0x0A>', '\n')
 
 # First model prediction ko-deplot
+@spaces.GPU(enable_queue=True)
 def predict_model1(image):
     images = [image]
     inputs = processor1(images=images, text="What is the title of the chart", return_tensors="pt", padding=True)
@@ -110,6 +109,7 @@ def replace_unk(text):
     return text
 
 # Second model prediction aihub_deplot
+@spaces.GPU(enable_queue=True)
 def predict_model2(image):
     image = image.convert("RGB")
     inputs = processor2(images=image, return_tensors="pt", max_patches=MAX_PATCHES).to(device)
@@ -126,6 +126,7 @@ def predict_model2(image):
     refined_table = replace_unk(generated_datatable)
     return refined_table
 
+@spaces.GPU(enable_queue=True)
 def predict_model3(image):
     image=image.convert("RGB")
     input_prompt = "<extract_data_table> <s_answer>"