Spaces:

Sangjun2
/

ultimate_gradio

Build error

Sangjun2 commited on Aug 24, 2024

Commit

b091eac

verified ·

1 Parent(s): 666d8c2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ processor2,model2=load_model2()
 #Load third model unichart
 def load_model3():
-    unichart_model_path = "./unichart2"
     model3 =  VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
     processor3 = DonutProcessor.from_pretrained(unichart_model_path)
     model3.to(torch.device("cuda"))

 #Load third model unichart
 def load_model3():
+    unichart_model_path = "./unichart3/chartqa-checkpoint-epoch=3-170000"
     model3 =  VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
     processor3 = DonutProcessor.from_pretrained(unichart_model_path)
     model3.to(torch.device("cuda"))