BhumikaMak commited on
Commit
d4cb7c6
·
1 Parent(s): 6c34a8c

Debug: refactor src

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  import torchvision.transforms as transforms
5
  import gradio as gr
6
  from yolov5 import xai_yolov5
7
- from yolov8 import xai_yolov8n
8
 
9
  def process_image(image, yolo_versions=["yolov5"]):
10
  image = np.array(image)
 
4
  import torchvision.transforms as transforms
5
  import gradio as gr
6
  from yolov5 import xai_yolov5
7
+ from yolov8 import xai_yolov8s
8
 
9
  def process_image(image, yolo_versions=["yolov5"]):
10
  image = np.array(image)