Update app.py

app.py

@@ -1,4 +1,3 @@
-#import tensorflow_addons as tfa
 import gradio as gr
 import tensorflow as tf
 import numpy as np
@@ -44,7 +43,7 @@ def classify_image(inp):
                 result[label] = float(predicted_class_indices[i])
             except KeyError:
                 print(f"KeyError: Label not found for index {predicted_class_indices[i]}")
-    return f"{prediction}..{predicted_class_indices}" 
+    return f"{prediction}...{prediction[0[2]]}..{predicted_class_indices}"