change map location to cpu

app.py

@@ -49,7 +49,7 @@ t5_model_path = './ke_t5.pt'
 def load_model1():
     processor1 = Pix2StructProcessor.from_pretrained('nuua/ko-deplot')
     model1 = Pix2StructForConditionalGeneration.from_pretrained('nuua/ko-deplot')
-    model1.load_state_dict(torch.load(ko_deplot_model_path, map_location=device))
+    model1.load_state_dict(torch.load(ko_deplot_model_path, map_location="cpu"))
     model1.to(device)
     return processor1,model1
 
@@ -60,7 +60,7 @@ processor1,model1=load_model1()
 def load_model2():
     processor2 = AutoProcessor.from_pretrained("ybelkada/pix2struct-base")
     model2 = Pix2StructForConditionalGeneration.from_pretrained("ybelkada/pix2struct-base")
-    model2.load_state_dict(torch.load(aihub_deplot_model_path, map_location=device))
+    model2.load_state_dict(torch.load(aihub_deplot_model_path, map_location="cpu"))
     model2.to(device)
     return processor2,model2