Spaces:

Sangjun2
/

skku-vaiv2

Runtime error

Sangjun2 commited on Dec 18, 2024

Commit

9272035

verified ·

1 Parent(s): 83663f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ ko_deplot_model_path = './deplot_model_ver_24.11.21_korean_only(exclude NUUA)_ep
 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

 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.to(torch.device("cuda"))
     return processor1, model1