maxidl commited on
Commit
c7d8cb8
1 Parent(s): 137c997
Files changed (2) hide show
  1. app.py +7 -0
  2. requirements.txt +3 -2
app.py CHANGED
@@ -3,6 +3,13 @@ import spaces
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
 
 
 
 
 
 
 
6
 
7
  model_name = "maxidl/arena-test"
8
 
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
6
+ from marker.convert import convert_single_pdf
7
+ from marker.output import markdown_exists, save_markdown, get_markdown_filepath
8
+ from marker.pdf.utils import find_filetype
9
+ from marker.pdf.extract_text import get_length_of_text
10
+ from marker.models import load_all_models
11
+ from marker.settings import settings
12
+ from marker.logger import configure_logging
13
 
14
  model_name = "maxidl/arena-test"
15
 
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
- transformers
2
- accelerate
 
 
1
+ transformers==4.45.2
2
+ accelerate
3
+ marker @ git+https://github.com/maxidl/marker-arena@main