george sami commited on
Commit
bafe32a
·
1 Parent(s): 3f3d3ac

add packgaes to requirements

Browse files
Files changed (5) hide show
  1. app.py +5 -2
  2. dog1.avif +0 -0
  3. dog1.jpeg +0 -0
  4. dog2.jpg +0 -0
  5. dog2.webp +0 -0
app.py CHANGED
@@ -7,14 +7,17 @@ learner = load_learner("model.pkl")
7
  categories = ("cat", "dog")
8
 
9
  def classify_image(img):
 
 
 
10
  pred, pred_idx, probs = learner.predict(img)
11
- return {categories[i]: float(probs[i]) for i in range(len(categories))}
12
 
13
  demo = gr.Interface(
14
  fn=classify_image,
15
  inputs=gr.Image(type="pil"),
16
  outputs=gr.Label(),
17
- examples=["cat1.jpeg", "dog1.avif", "cat2.jpeg", "dog2.webp"],
18
  )
19
 
20
  demo.launch()
 
7
  categories = ("cat", "dog")
8
 
9
  def classify_image(img):
10
+ if img is None:
11
+ return {}
12
+ img = PILImage.create(img)
13
  pred, pred_idx, probs = learner.predict(img)
14
+ return dict(zip(categories, map(float, probs)))
15
 
16
  demo = gr.Interface(
17
  fn=classify_image,
18
  inputs=gr.Image(type="pil"),
19
  outputs=gr.Label(),
20
+ examples=["cat1.jpeg", "dog1.jpeg", "cat2.jpeg", "dog2.jpg"],
21
  )
22
 
23
  demo.launch()
dog1.avif DELETED
Binary file (59.6 kB)
 
dog1.jpeg ADDED
dog2.jpg ADDED
dog2.webp DELETED
Binary file (98.5 kB)