Pranomvignesh commited on
Commit
f2a6c90
·
1 Parent(s): 419e1eb

Modified examples

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -8,6 +8,7 @@ imageClassifier = pipeline(task="image-classification",
8
 
9
  model = yolov5.load('./gentle-meadow.pt', device="cpu")
10
 
 
11
  def predict(image):
12
  # results = model([image], size=224)
13
  predictions = imageClassifier(image)
@@ -21,14 +22,14 @@ description = """
21
  Try the examples at bottom to get started.
22
  """
23
  examples = [
24
- [os.path.join(os.path.abspath(''), './examples/sample_1.png')],
25
- [os.path.join(os.path.abspath(''), './examples/sample_2.png')],
26
- [os.path.join(os.path.abspath(''), './examples/sample_3.jpg')],
27
- [os.path.join(os.path.abspath(''), './examples/sample_4.jpg')],
28
- [os.path.join(os.path.abspath(''), './examples/sample_5.jpg')],
29
- [os.path.join(os.path.abspath(''), './examples/sample_6.jpg')],
30
- [os.path.join(os.path.abspath(''), './examples/sample_7.jpg')],
31
- [os.path.join(os.path.abspath(''), './examples/sample_8.jpg')],
32
  ]
33
 
34
  inputs = gr.Image(type="pil", shape=(224, 224),
 
8
 
9
  model = yolov5.load('./gentle-meadow.pt', device="cpu")
10
 
11
+
12
  def predict(image):
13
  # results = model([image], size=224)
14
  predictions = imageClassifier(image)
 
22
  Try the examples at bottom to get started.
23
  """
24
  examples = [
25
+ [os.path.abspath('./examples/sample_1.png')],
26
+ [os.path.abspath('./examples/sample_2.png')],
27
+ [os.path.abspath('./examples/sample_3.jpg')],
28
+ [os.path.abspath('./examples/sample_4.jpg')],
29
+ [os.path.abspath('./examples/sample_5.jpg')],
30
+ [os.path.abspath('./examples/sample_6.jpg')],
31
+ [os.path.abspath('./examples/sample_7.jpg')],
32
+ [os.path.abspath('./examples/sample_8.jpg')],
33
  ]
34
 
35
  inputs = gr.Image(type="pil", shape=(224, 224),