File size: 2,628 Bytes
4e35e2b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
setup: install preprocess augment

setup-runpod: install-runpod preprocess augment

install:
	pip3 install --no-cache-dir --break-system-packages -r requirements-cpu.txt
	pip3 install --no-cache-dir --break-system-packages -r requirements.txt

install-runpod:
	pip3 install --break-system-packages --ignore-installed blinker -r requirements.txt
	pip3 install --break-system-packages --force-reinstall "torch>=2.6.0" "torchvision>=0.21.0"
	pip3 install --break-system-packages --force-reinstall "numpy<2"

install-mac:
	/usr/local/bin/python3 -m pip install --break-system-packages -r requirements.txt

preprocess:
	python3 -m src.models.preprocess

augment:
	python3 -m src.models.augment --max-length $(or $(MAX_LENGTH),256)

serialize:
	python3 -m src.models.serialize --model-dir $(MODEL_DIR)

serialize-marker:
	python3 -m src.models.serialize --model-dir models/marker

serialize-qa-m:
	python3 -m src.models.serialize --model-dir models/qa_m

serialize-qa-b:
	python3 -m src.models.serialize --model-dir models/qa_b

serialize-all: serialize-marker serialize-qa-m serialize-qa-b

inference:
	python3 -m src.models.inference --model-dir $(MODEL_DIR) --data $(DATA) --output $(OUTPUT)

inference-marker:
	python3 -m src.models.inference --model-dir models/marker --data $(DATA) --output predictions_marker.json

inference-qa-m:
	python3 -m src.models.inference --model-dir models/qa_m --data $(DATA) --output predictions_qa_m.json

inference-qa-b:
	python3 -m src.models.inference --model-dir models/qa_b --data $(DATA) --output predictions_qa_b.json

inference-all: inference-marker inference-qa-m inference-qa-b

hf-upload:
	python3 -m src.models.hf_upload $(if $(MODE),--mode $(MODE))

hf-upload-marker:
	python3 -m src.models.hf_upload --mode marker

hf-upload-qa-m:
	python3 -m src.models.hf_upload --mode qa_m

hf-upload-qa-b:
	python3 -m src.models.hf_upload --mode qa_b

hf-upload-fasttext:
	python3 -m src.models.hf_upload --mode fasttext

hf-upload-all: hf-upload-marker hf-upload-qa-m hf-upload-qa-b hf-upload-fasttext

hf-download:
	python3 -m src.models.hf_download $(if $(MODE),--mode $(MODE))

hf-download-marker:
	python3 -m src.models.hf_download --mode marker

hf-download-qa-m:
	python3 -m src.models.hf_download --mode qa_m

hf-download-qa-b:
	python3 -m src.models.hf_download --mode qa_b

hf-download-fasttext:
	python3 -m src.models.hf_download --mode fasttext

hf-download-all: hf-download-marker hf-download-qa-m hf-download-qa-b hf-download-fasttext

train-fasttext:
	python3 -m src.models.fasttext

start-be:
	uvicorn app:app --host 0.0.0.0 --port 8000 --reload

start-fe:
	streamlit run main.py