Update app.py

app.py

@@ -16,6 +16,14 @@ models = {
     'UNet3+': torch.jit.load('./model/UNet3plus.pt'),
 }
 
+# Função para ordenar e pré-processar a imagem de entrada
+def ordenar_e_preprocessar_imagem(img, modelo):
+    ori = img.copy()
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    h, w = img.shape
+    img_out = preprocessar_imagem(img, modelo)
+    return img_out, h, w, img, ori
+    
 def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
     try:
         # Faça uma cópia da imagem original