nehulagrawal commited on
Commit
64d1f9a
·
1 Parent(s): 4f32028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -5,9 +5,8 @@ from sahi.utils.cv import visualize_object_predictions, read_image
5
  from ultralyticsplus import YOLO, render_result
6
 
7
  # Images
8
- torch.hub.download_url_to_file('https://huggingface.co/spaces/foduucom/table-extraction-yolov8/resolve/main/test/table1.jpg', 'document1.jpg')
9
- torch.hub.download_url_to_file('https://huggingface.co/spaces/foduucom/table-extraction-yolov8/resolve/main/test/table2.jpg', 'document2.jpg')
10
- torch.hub.download_url_to_file('https://huggingface.co/spaces/foduucom/table-extraction-yolov8/resolve/main/test/table3.jpg', 'document3.jpg')
11
 
12
  def yolov8_inference(
13
  image: gr.inputs.Image = None,
 
5
  from ultralyticsplus import YOLO, render_result
6
 
7
  # Images
8
+ torch.hub.download_url_to_file('https://huggingface.co/spaces/foduucom/table-extraction-yolov8/resolve/main/samples/1.jpeg', '1.jpeg')
9
+ torch.hub.download_url_to_file('https://huggingface.co/spaces/foduucom/table-extraction-yolov8/resolve/main/samples/2.jpg', '2.JPG')
 
10
 
11
  def yolov8_inference(
12
  image: gr.inputs.Image = None,