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