Update app.py

app.py

@@ -31,11 +31,10 @@ loaded_model.load_weights('modelo_treinado.h5')
 
 # Função para fazer previsões usando o modelo treinado
 def predict_image(input_image):
-    # Realize o pré-processamento na imagem de entrada, se necessário
-    # input_image = preprocess_image(input_image)
+    # Realize o pré-processamento na imagem de entrada
+    input_image = preprocess_image(input_image)
 
     # Faça uma previsão usando o modelo carregado
-    input_image = tf.image.resize(input_image, (IMG_HEIGHT, IMG_WIDTH))
     input_image = tf.expand_dims(input_image, axis=0)
     prediction = loaded_model.predict(input_image)
 
@@ -48,6 +47,7 @@ def predict_image(input_image):
     formatted_text = f"Predicted Class: {predicted_class}\nProbability: {probability:.2%}"
     return formatted_text
 
+
 # Crie uma interface Gradio para fazer previsões
 iface = gr.Interface(
     fn=predict_image,