Spaces:
Runtime error
Runtime error
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()
|