Update app.py

app.py

@@ -42,13 +42,18 @@ MAX_PATCHES = 512
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Paths to the models
-ko_deplot_model_path = './deplot_model_ver_24.11.21_korean_only(exclude NUUA)_epoch1.bin'
+#ko_deplot_model_path = './deplot_model_ver_24.11.21_korean_only(exclude NUUA)_epoch1.bin'
+
+file_path = hf_hub_download(
+    repo_id="Sangjun2/skku-Deplot", 
+    filename="deplot_model_ver_24.11.21_korean_only(exclude NUUA)_epoch3.bin"
+)
 
 # Load first model ko-deplot
 def load_model1():
     processor1 = Pix2StructProcessor.from_pretrained('nuua/ko-deplot')
     model1 = Pix2StructForConditionalGeneration.from_pretrained('nuua/ko-deplot')
-    model1.load_state_dict(torch.load('Sangjun2/skku-Deplot/deplot_model_ver_24.11.21_korean_only(exclude NUUA)_epoch3.bin', map_location="cpu"))
+    model1.load_state_dict(torch.load(file_path, map_location="cpu"))
     model1.to(torch.device("cuda"))
     return processor1, model1