add gpu decorator

app.py

@@ -47,20 +47,26 @@ t5_model_path = './ke_t5.pt'
 # Load first model ko-deplot
 processor1 = Pix2StructProcessor.from_pretrained('nuua/ko-deplot')
 model1 = Pix2StructForConditionalGeneration.from_pretrained('nuua/ko-deplot')
+@spaces.GPU(enable_queue=True)
 model1.load_state_dict(torch.load(ko_deplot_model_path, map_location=device))
+@spaces.GPU(enable_queue=True)
 model1.to(device)
 
 # Load second model aihub-deplot
 processor2 = AutoProcessor.from_pretrained("ybelkada/pix2struct-base")
 model2 = Pix2StructForConditionalGeneration.from_pretrained("ybelkada/pix2struct-base")
+@spaces.GPU(enable_queue=True)
 model2.load_state_dict(torch.load(aihub_deplot_model_path, map_location=device))
 
 
 tokenizer = T5Tokenizer.from_pretrained("KETI-AIR/ke-t5-base")
 t5_model = T5ForConditionalGeneration.from_pretrained("KETI-AIR/ke-t5-base")
+@spaces.GPU(enable_queue=True)
 t5_model.load_state_dict(torch.load(t5_model_path, map_location=device))
 
+@spaces.GPU(enable_queue=True)
 model2.to(device)
+@spaces.GPU(enable_queue=True)
 t5_model.to(device)
 
 #Load third model unichart
@@ -68,6 +74,7 @@ unichart_model_path = "./unichart"
 model3 =  VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
 processor3 = DonutProcessor.from_pretrained(unichart_model_path)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+@spaces.GPU(enable_queue=True)
 model3.to(device)  
 
 #ko-deplot 추론함수