hanquansanren commited on
Commit
f312178
·
1 Parent(s): 15c8a9e
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,7 +36,7 @@ from datasets.utils.warping import register_model2
36
  import gradio as gr
37
  from huggingface_hub import hf_hub_download
38
 
39
-
40
 
41
  reg_model_bilin = register_model2((512,512), 'bilinear')
42
 
@@ -370,6 +370,7 @@ if __name__ == '__main__':
370
  ],
371
  title="Document Image Dewarping",
372
  description="This is a demo for document image dewarping using a trained model.",
 
373
  )
374
 
375
  # demo.launch(share=True, debug=True, server_name="10.7.88.77")
 
36
  import gradio as gr
37
  from huggingface_hub import hf_hub_download
38
 
39
+ example_img_list = ['3_2 copy.png', '25_1 copy.png']
40
 
41
  reg_model_bilin = register_model2((512,512), 'bilinear')
42
 
 
370
  ],
371
  title="Document Image Dewarping",
372
  description="This is a demo for document image dewarping using a trained model.",
373
+ examples=example_img_list
374
  )
375
 
376
  # demo.launch(share=True, debug=True, server_name="10.7.88.77")