ooferdoodles commited on
Commit
98913b7
β€’
1 Parent(s): 9b8e51b

i want die

Browse files
app.py CHANGED
@@ -12,11 +12,12 @@ learn = load_learner(Path("model.pkl"))
12
 
13
  def classify_image(img):
14
  pred, idx, probs = learn.predict(img)
 
15
  return f"This is {pred}.\nConfidence Level: {float(max(probs))}"
16
 
17
 
18
  image = gr.Image(shape=(192, 192))
19
- examples = ['pizza.jpg', 'pasta.jpg', 'dunno.jpg']
20
 
21
  app = gr.Interface(fn=classify_image,
22
  inputs=image,
 
12
 
13
  def classify_image(img):
14
  pred, idx, probs = learn.predict(img)
15
+
16
  return f"This is {pred}.\nConfidence Level: {float(max(probs))}"
17
 
18
 
19
  image = gr.Image(shape=(192, 192))
20
+ examples = ['images/pizza.jpg', 'images/pasta.jpg', 'images/dunno.jpg']
21
 
22
  app = gr.Interface(fn=classify_image,
23
  inputs=image,
dunno.jpg β†’ images/dunno.jpg RENAMED
File without changes
pasta.jpg β†’ images/pasta.jpg RENAMED
File without changes
pizza.jpg β†’ images/pizza.jpg RENAMED
File without changes