Update app.py

app.py

@@ -2,10 +2,17 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 from PIL import Image
+import json
 
 # Carregue o modelo previamente treinado
 model = tf.keras.models.load_model("model.h5")
 
+# Carregue o arquivo JSON com as categorias indexadas
+with open("categories.json", "r") as json_file:
+    categories_data = json.load(json_file)
+
+categories = [entry["category"] for entry in categories_data]
+
 # Crie uma função para realizar a classificação
 def classify_image(image):
     try:
@@ -19,7 +26,7 @@ def classify_image(image):
 
         # Decodifique a classe prevista
         class_idx = np.argmax(prediction)
-        class_label = categories[class_idx]  # Substitua 'categories' pela sua lista de classes
+        class_label = categories[class_idx]
 
         return class_label
     except Exception as e: