Sangjun2 commited on
Commit
ab8d6b9
·
verified ·
1 Parent(s): a27602a

remove gpu decorator located in load_model methods

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ def load_model1():
55
  processor1,model1=load_model1()
56
 
57
  # Load second model aihub-deplot
58
- @spaces.GPU(enable_queue=True,duration=120)
59
  def load_model2():
60
  processor2 = AutoProcessor.from_pretrained("ybelkada/pix2struct-base")
61
  model2 = Pix2StructForConditionalGeneration.from_pretrained("ybelkada/pix2struct-base")
@@ -67,7 +67,7 @@ processor2,model2=load_model2()
67
 
68
 
69
  #Load third model unichart
70
- @spaces.GPU(enable_queue=True,duration=120)
71
  def load_model3():
72
  unichart_model_path = "./unichart"
73
  model3 = VisionEncoderDecoderModel.from_pretrained(unichart_model_path)
 
55
  processor1,model1=load_model1()
56
 
57
  # Load second model aihub-deplot
58
+
59
  def load_model2():
60
  processor2 = AutoProcessor.from_pretrained("ybelkada/pix2struct-base")
61
  model2 = Pix2StructForConditionalGeneration.from_pretrained("ybelkada/pix2struct-base")
 
67
 
68
 
69
  #Load third model unichart
70
+
71
  def load_model3():
72
  unichart_model_path = "./unichart"
73
  model3 = VisionEncoderDecoderModel.from_pretrained(unichart_model_path)