BhumikaMak commited on
Commit
9b2c5e1
·
1 Parent(s): c2d0764

Fix: argument mismatch

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import numpy as np
2
  import cv2
 
3
  from PIL import Image
4
  import torchvision.transforms as transforms
5
  import gradio as gr
@@ -34,13 +35,12 @@ interface = gr.Interface(
34
  title="Explainable AI for YOLO Models",
35
  description="Upload an image or select a sample to visualize YOLO object detection with Grad-CAM.",
36
  examples=[
37
- ["/data/xai/sample1.jpeg"],
38
- ["/data/xai/sample2.jpeg"],
39
  ],
40
  live=True
41
  )
42
 
43
 
44
-
45
  if __name__ == "__main__":
46
  interface.launch()
 
1
  import numpy as np
2
  import cv2
3
+ import os
4
  from PIL import Image
5
  import torchvision.transforms as transforms
6
  import gradio as gr
 
35
  title="Explainable AI for YOLO Models",
36
  description="Upload an image or select a sample to visualize YOLO object detection with Grad-CAM.",
37
  examples=[
38
+ [os.path.join(os.getcwd(),"/data/xai/sample1.jpeg")],
39
+ [os.path.join(os.getcwd(), "/data/xai/sample2.jpeg")],
40
  ],
41
  live=True
42
  )
43
 
44
 
 
45
  if __name__ == "__main__":
46
  interface.launch()