aicloud
/
object_detection
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

This commit is contained in:
SOULOFCINDER 2023-03-31 15:59:14 +08:00
parent ec104e3048
commit f73e8dff96
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
table_transformer_detection/app.py
View File

@ -1,7 +1,6 @@
import gradio as gr import gradio as gr
from transformers import AutoImageProcessor, TableTransformerForObjectDetection from transformers import AutoImageProcessor, TableTransformerForObjectDetection
import torch import torch
from PIL import Image
def inference(img): def inference(img):
pretrained_model_path = "table-transformer-detection" pretrained_model_path = "table-transformer-detection"

1
table_transformer_detection/requirements.txt
View File

@ -1,2 +1,3 @@
gradio==3.21.0 gradio==3.21.0
transformers==4.27.1 transformers==4.27.1
torch==2.0.0