From f73e8dff96f8cd439a54f2103ace92aac68e3742 Mon Sep 17 00:00:00 2001 From: SOULOFCINDER <841135647@qq.com> Date: Fri, 31 Mar 2023 15:59:14 +0800 Subject: [PATCH] update --- table_transformer_detection/app.py | 1 - table_transformer_detection/requirements.txt | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/table_transformer_detection/app.py b/table_transformer_detection/app.py index 410ead1..8157f30 100644 --- a/table_transformer_detection/app.py +++ b/table_transformer_detection/app.py @@ -1,7 +1,6 @@ import gradio as gr from transformers import AutoImageProcessor, TableTransformerForObjectDetection import torch -from PIL import Image def inference(img): pretrained_model_path = "table-transformer-detection" diff --git a/table_transformer_detection/requirements.txt b/table_transformer_detection/requirements.txt index 973b15c..3244b33 100644 --- a/table_transformer_detection/requirements.txt +++ b/table_transformer_detection/requirements.txt @@ -1,2 +1,3 @@ gradio==3.21.0 transformers==4.27.1 +torch==2.0.0