Update app.py

app.py

@@ -42,10 +42,12 @@ def clf(name):
     data[0] = normalized_image_array
 
     # run the inference
-    prediction = model.predict(data)
-    class_index = np.argmax(prediction)
+    prediction = model.predict(data)    
+    confidences = {label_dict[i]: float(prediction[i]) for i in range(4)}    
     
-    return label_dict[class_index]
+    return confidences
 
-iface = gr.Interface(fn=clf, inputs="image", outputs="text")
-iface.launch()
+gr.Interface(fn=predict, 
+             inputs=gr.Image(type="pil"),
+             outputs=gr.Label(num_top_classes=5),
+             examples=["12122.jpg", "12342.jpg", "23482.jpg", "23992.jpg", "29349.jpg", "29384.jpg", "32321.jpg", "43413.jpg", "59394.jpg", "77923.jpg", "83823.jpg", "88893.jpg", "88921.jpg", "88923.jpg", "92309.jpg", "92921.jpg"]).launch()