Spaces:
Running
Running
saylee-m
commited on
Commit
·
59e7947
1
Parent(s):
e0a50f9
pg enabled
Browse files
app.py
CHANGED
@@ -45,11 +45,11 @@ def load_models():
|
|
45 |
donut_model, donut_processor = load_donut_model()
|
46 |
print("donut downloaded")
|
47 |
# #load paligemma
|
48 |
-
|
49 |
-
|
50 |
|
51 |
return {"donut":[donut_model, donut_processor],
|
52 |
-
|
53 |
}
|
54 |
|
55 |
loaded_models = load_models()
|
|
|
45 |
donut_model, donut_processor = load_donut_model()
|
46 |
print("donut downloaded")
|
47 |
# #load paligemma
|
48 |
+
pg_model, pg_processor = load_paligemma_docvqa()
|
49 |
+
print("paligemma downloaded")
|
50 |
|
51 |
return {"donut":[donut_model, donut_processor],
|
52 |
+
"paligemma": [pg_model, pg_processor]
|
53 |
}
|
54 |
|
55 |
loaded_models = load_models()
|