Update app.py

app.py

@@ -50,6 +50,7 @@ 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="cpu"))
+    model1.to(torch.device("cuda"))
     return processor1,model1
 
 processor1,model1=load_model1()