Beansbeansbeansbeans commited on
Commit
74ffc74
·
verified ·
1 Parent(s): f16e48f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -15,6 +15,9 @@ metal_model = from_pretrained_fastai("pyesonekyaw/recycletree_metal")
15
  others_model = from_pretrained_fastai("pyesonekyaw/recycletree_others")
16
  glass_model = from_pretrained_fastai("pyesonekyaw/recycletree_glass")
17
 
 
 
 
18
  material_names = ['Glass', 'Metal', 'Others', 'Paper', 'Plastic']
19
  plastic_names = ['CD Disk', 'Straw', 'Plastic Bag', 'Clothes Hanger', 'Plastic Container or Bottle',
20
  'Disposable Cutlery', 'Plastic Packaging', 'Plastic Packaging With Foil', 'Styrofoam']
@@ -221,6 +224,7 @@ with gr.Blocks(title="Trash Classification", css="#custom_header {min-height: 3r
221
  item_probs = gr.outputs.Label(label="Item Prediction")
222
 
223
  gr.Examples(
 
224
  inputs=input_image,
225
  fn=predict_image,
226
  cache_examples=False,
 
15
  others_model = from_pretrained_fastai("pyesonekyaw/recycletree_others")
16
  glass_model = from_pretrained_fastai("pyesonekyaw/recycletree_glass")
17
 
18
+ examples = ["Examples/1.jpg", "Examples/2.jpg",
19
+ "Examples/3.jpg", "Examples/4.jpg", "Examples/5.jpg"]
20
+
21
  material_names = ['Glass', 'Metal', 'Others', 'Paper', 'Plastic']
22
  plastic_names = ['CD Disk', 'Straw', 'Plastic Bag', 'Clothes Hanger', 'Plastic Container or Bottle',
23
  'Disposable Cutlery', 'Plastic Packaging', 'Plastic Packaging With Foil', 'Styrofoam']
 
224
  item_probs = gr.outputs.Label(label="Item Prediction")
225
 
226
  gr.Examples(
227
+ examples=examples,
228
  inputs=input_image,
229
  fn=predict_image,
230
  cache_examples=False,