Update processar_imagem_de_entrada.py

processar_imagem_de_entrada.py

@@ -1,15 +1,21 @@
 import os
-import time
+import logging
 import torch
-import numpy as np
 import cv2
+import numpy as np
+from preprocess import unsharp_masking
+import time
+from sklearn.cluster import KMeans
+import os
+from preprocessar_imagem import preprocessar_imagem
 
-# Função para processar a imagem de entrada
-import logging
-logging.basicConfig(level=logging.DEBUG)  # Enable logging
+# Define the "device" variable at the global scope
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# ... (rest of your code)
 
 # Função para processar a imagem de entrada
-def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
+def processar_imagem_de_entrada(img, modelo, pipe, device, salvar_resultado=False):
     try:
         # Faça uma cópia da imagem original
         img = img.copy()
@@ -51,7 +57,7 @@ def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
         return spent, logit
     
     except Exception as e:
-        # Em caso de erro, retorne uma mensagem de erro
-        logging.error(f"Error: {str(e)}")  # Log the error message
+        logging.error(f"Error: {str(e)}")
         return str(e), None
 
+# ... (rest of your code)