Update processar_imagem_de_entrada.py

processar_imagem_de_entrada.py

@@ -4,6 +4,7 @@ import torch
 import numpy as np
 import cv2
 
+# Função para processar a imagem de entrada
 # Função para processar a imagem de entrada
 def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
     try:
@@ -43,10 +44,10 @@ def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
         # Salve o resultado em um arquivo se a opção estiver ativada
         if salvar_resultado:
             nome_arquivo = os.path.join(caminho_salvar_resultado, f'resultado_{int(time.time())}.png')
-            cv2.imwrite(nome_arquivo, img_out)
-        
-        return spent, img_out
+            cv2.imwrite(nome_arquivo, logit)  # Corrected variable name
+            
+        return spent, logit  # Corrected variable name
     
     except Exception as e:
         # Em caso de erro, retorne uma mensagem de erro
-        return str(e), None
+        return str(e), None