Sangjun2 commited on
Commit
14d9e42
·
verified ·
1 Parent(s): 330246e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ processor2,model2=load_model2()
70
  #Load third model unichart
71
 
72
  def load_model3():
73
- unichart_model_path = "./unichart"
74
  model3 = VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
75
  processor3 = DonutProcessor.from_pretrained(unichart_model_path)
76
  model3.to(torch.device("cuda"))
 
70
  #Load third model unichart
71
 
72
  def load_model3():
73
+ unichart_model_path = "./unichart2"
74
  model3 = VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
75
  processor3 = DonutProcessor.from_pretrained(unichart_model_path)
76
  model3.to(torch.device("cuda"))