Update app.py

app.py

@@ -49,7 +49,7 @@ def classify_image(inp):
     inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
     inp = tf.keras.applications.vgg16.preprocess_input(inp)
     prediction = model.predict(inp).flatten()
-    return {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}
+    return {labels[i]: f"{prediction[i]:.6f}" for i in range(NUM_CLASSES)}
 
 # image = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))
 # label = gr.outputs.Label(num_top_classes=2)