Update app.py.py

app.py.py

@@ -6,17 +6,13 @@ Automatically generated by Colaboratory.
 Original file is located at
     https://colab.research.google.com/drive/13X2E9v7GxryXyT39R5CzxrNwxfA6KMFJ
 """
-
-!pip install gradio
-
 import gradio as gr
 from PIL import Image
 from timeit import default_timer as timer
 from tensorflow import keras
 import numpy as np
 
-MODEL = keras.models.load_model(
-            "convnet_from_scratch_with_augmentation.keras")
+MODEL = keras.models.load_model("convnet_from_scratch_with_augmentation.keras")
 
 def predict(img):
 
@@ -41,6 +37,7 @@ def predict(img):
 predict('/content/cat.1505.jpg')
 
 # Create title, description and article strings
+example_list = [["examples/" + example] for example in os.listdir("examples")]
 title = "Classification Demo"
 description = "Cat/Dog classification Tensorflow model with Augmentted small dataset"
 
@@ -49,6 +46,7 @@ demo = gr.Interface(fn=predict, # mapping function from input to output
                     inputs=gr.Image(type='filepath'), # what are the inputs?
                     outputs=[gr.Label(label="Predictions"), # what are the outputs?
                              gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
+                    examples=example_list,
                     title=title,
                     description=description,)
 
@@ -56,15 +54,3 @@ demo = gr.Interface(fn=predict, # mapping function from input to output
 demo.launch(debug=False, # print errors locally?
             share=True) # generate a publically shareable URL?
 
-pip install tensorflow
-
-import PIL
-
-import tensorflow as tf
-
-import timeit
-
-print(gr.__version__)
-print(np.__version__)
-print(tf.__version__)
-print(PIL.__version__)