Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from tensorflow.keras.models import load_model
+import cv2
 
 # Função para fazer a previsão de classificação
 def classify_pcos(image):
@@ -16,11 +17,20 @@ def classify_pcos(image):
     return class_label
 
 # Função para pré-processar a imagem (redimensionar, normalizar, etc.)
+
 def preprocess_image(image):
-    # Implemente o pré-processamento necessário para a imagem
+    # Redimensionar a imagem para 224x224
+    resized_image = cv2.resize(image, (224, 224))
+    
+    # Normalizar a imagem (opcional, dependendo do pré-processamento necessário)
+    normalized_image = resized_image / 255.0
+    
+    # Adicionar dimensão extra para se adequar ao formato de entrada do modelo
+    preprocessed_image = np.expand_dims(normalized_image, axis=0)
     
     return preprocessed_image
 
+
 # Carregar o modelo treinado
 model = load_model('FightOS_CNN_Models.h5')