Update app.py

app.py

@@ -2,7 +2,6 @@ import tensorflow as tf
 from keras.utils import custom_object_scope
 import gradio as gr
 from PIL import Image, ImageDraw
-import numpy as np 
 
 # Defina a camada personalizada FixedDropout
 class FixedDropout(tf.keras.layers.Dropout):
@@ -35,7 +34,10 @@ def classify_image(inp):
     # Obtém a classe prevista
     predicted_class = class_names[np.argmax(prediction)]
 
-    return inp, predicted_class
+    # Redimensione a imagem de entrada para um tamanho menor (128x128)
+    inp = Image.fromarray(inp).resize((128, 128))
+
+    return np.array(inp), predicted_class
 
 # Crie uma interface Gradio
 iface = gr.Interface(