Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 
 def classify(img):
-  pred, probability = predict_one_image(img, model_VGG19)
+    pred, probability = predict_one_image(img, model_VGG19)
+    return class_names[pred]
 
 
 demo = gr.Interface(fn=classify, 
              inputs = gr.inputs.Image(shape=(224,224)), 
-             outputs =gr.outputs.Label(num_classes=5),
+             outputs =gr.outputs.Textbox(),
              title = "Flower Recognition",
              examples =['tulip.jpg','rose.jpg','daisy.jpg','dandelion.jpg','sunflower.jpg',],
              description = "Transfer Learning Application",