Hugging Face's logo

Spaces:
sadickam
/
PDF-text-extra
Runtime error

App Files Community
sadickam commited on
Commit
9df2f2e
·
verified ·
1 Parent(s): 2c73b80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
46
  model_1 = lp.Detectron2LayoutModel(
47
  'lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config',
48
  extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8],
49
- device=device
50
 
51
  def pdf_to_images(pdf_path, start_page=0, end_page=None):
52
  """Convert PDF pages to images."""
 
46
  model_1 = lp.Detectron2LayoutModel(
47
  'lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config',
48
  extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8],
49
+ device=device)
50
 
51
  def pdf_to_images(pdf_path, start_page=0, end_page=None):
52
  """Convert PDF pages to images."""