Update app.py

app.py

@@ -2,6 +2,7 @@ from PIL import Image
 import torch
 import gradio as gr
 from torchvision.transforms import Compose, Normalize, ToTensor, Resize, CenterCrop
+from pathlib import Path
 
 labels = {0: 'glass',
  1: 'metal',
@@ -13,6 +14,9 @@ labels = {0: 'glass',
 inference = torch.load('fine_tune_resnet.pth', map_location=torch.device('cpu'))
 inference.eval()
 
+
+example = [Path.cwd() / 'cardboard.wp', Path.cwd() / 'plastic.jpg']
+
 def classifier(image):
     test_transform = Compose([
             Resize(256),
@@ -32,5 +36,7 @@ def classifier(image):
 
 iface = gr.Interface(fn=classifier,
                     inputs=gr.Image(type="pil"),
-                    outputs='label')
+                    outputs='label',
+                    examples = example,
+                    title = 'Garbage Image Classification')
 iface.launch(share=True)