offthewallace commited on
Commit
da4e9c4
·
1 Parent(s): e043349

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  import torch
3
  from PIL import Image
4
-
 
5
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
6
 
7
 
 
1
  import gradio as gr
2
  import torch
3
  from PIL import Image
4
+ import pdf2image
5
+ import numpy as np
6
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
7
 
8