Spaces:
Build error
Build error
remove gpu decorator located in load_model methods
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def load_model1():
|
|
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,7 +67,7 @@ processor2,model2=load_model2()
|
|
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)
|
|
|
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)
|