Ubuntu commited on
Commit
a785cd5
·
1 Parent(s): 1b92647

added test cases

Browse files
app.py CHANGED
@@ -14,5 +14,8 @@ def detect_hotspots(image):
14
  return Image.fromarray(im_array[..., ::-1])
15
 
16
 
17
- demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(type='pil'), outputs="image", title="YOLO Object Detection")
 
 
 
18
  demo.launch()
 
14
  return Image.fromarray(im_array[..., ::-1])
15
 
16
 
17
+ demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(type='pil'), outputs="image", title="YOLO Object Detection",
18
+ examples=[['Test Case 1', 'images/test_image_1.jpg'], ['Test Case 2', 'images/test_image_2.jpg'],
19
+ ['Test Case 3', 'images/test_image_3.jpg'], ['Test Case 4', 'images/test_image_4.jpg']],
20
+ cache_examples= False,)
21
  demo.launch()
images/test_image_1.jpg ADDED
images/test_image_2.jpg ADDED
images/test_image_3.jpg ADDED
images/test_image_4.jpg ADDED