valiyevfagan commited on
Commit
2ec297b
·
verified ·
1 Parent(s): 8ad8dd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -11,6 +11,8 @@ labels = learn.dls.vocab
11
  def classify_trash(img):
12
  pred_class, pred_idx, probs = learn.predict(img)
13
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 
14
 
15
  # Gradio Interface
16
  interface = gr.Interface(
@@ -18,7 +20,8 @@ interface = gr.Interface(
18
  inputs=gr.Image(type="pil"),
19
  outputs=gr.Label(num_top_classes=5),
20
  title="Trash Classifier",
21
- description="Upload a trash image to classify it into one of 5 categories."
 
22
  )
23
 
24
  # Launch app
 
11
  def classify_trash(img):
12
  pred_class, pred_idx, probs = learn.predict(img)
13
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
14
+
15
+ examples = ["glass.png","plastic.jpg"]
16
 
17
  # Gradio Interface
18
  interface = gr.Interface(
 
20
  inputs=gr.Image(type="pil"),
21
  outputs=gr.Label(num_top_classes=5),
22
  title="Trash Classifier",
23
+ description="Upload a trash image to classify it into one of 5 categories.",
24
+ examples=examples
25
  )
26
 
27
  # Launch app