Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

streamlit app using Table Transformer and OCR #200

Open
wants to merge 16 commits into
base: master
Choose a base branch
from
Open

streamlit app using Table Transformer and OCR #200

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
7778a78
streamlit app using Table Transformer and OCR
salman-moh Oct 19, 2022
c3b4228
updated streamlit app using Table Transformer + OCR
salman-moh Oct 19, 2022
c210ab4
Updated README to include HF space link
salman-moh Oct 19, 2022
a6b0b54
Update Table Transformer/app.py
salman-moh Oct 19, 2022
12c4a81
Update Table Transformer/app.py
salman-moh Oct 19, 2022
5e7e9f8
Table Transformer details added
salman-moh Oct 19, 2022
1474d2f
Update README.md
salman-moh Oct 19, 2022
2226e5c
Update README.md
salman-moh Oct 19, 2022
58571e4
updated case for when there is no table
salman-moh Oct 19, 2022
aeef89b
Update app.py
salman-moh Oct 19, 2022
2f9623f
reflects HF Space streamlit app
salman-moh Oct 19, 2022
2b5bce3
Update README.md
salman-moh Oct 22, 2022
4922aa4
Deleted to have this repo as notebook-only
salman-moh Oct 22, 2022
3c915df
uploaded demo screenshot
salman-moh Oct 22, 2022
cffc6d6
Update Table Transformer/README.md
salman-moh Oct 22, 2022
c382285
Update README.md
salman-moh Oct 23, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions Table Transformer/app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ def table_detector(image, THRESHOLD_PROBA):
feature_extractor = DetrFeatureExtractor(do_resize=True, size=800, max_size=800)
encoding = feature_extractor(image, return_tensors="pt")

model = DetrForObjectDetection.from_pretrained("SalML/DETR-table-detection")
model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-detection")
salman-moh marked this conversation as resolved.
Show resolved Hide resolved

with torch.no_grad():
outputs = model(**encoding)
Expand All @@ -163,7 +163,7 @@ def table_struct_recog(image, THRESHOLD_PROBA):
feature_extractor = DetrFeatureExtractor(do_resize=True, size=1000, max_size=1000)
encoding = feature_extractor(image, return_tensors="pt")

model = DetrForObjectDetection.from_pretrained("SalML/DETR-table-structure-recognition")
model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-structure-recognition")
salman-moh marked this conversation as resolved.
Show resolved Hide resolved
with torch.no_grad():
outputs = model(**encoding)

Expand Down