saylee-m commited on
Commit
59e7947
·
1 Parent(s): e0a50f9

pg enabled

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # 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()
 
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()