Salvando alterações

Inferencia.py

@@ -1,6 +1,7 @@
 from transformers import pipeline
 import torch
 import json
+from LogCustomizado import LogCustomizado as logc
 
 class Inferencia:
 
@@ -8,6 +9,7 @@ class Inferencia:
         ...
 
     def extrair_dados_imagem(self, prompt, imagem):
+        logc.gerar_log(mensagem_log="EXECUTANDO PIPELINE...")
         pipe = pipeline(
             "image-text-to-text",
             model="google/gemma-3-4b-it",
@@ -28,6 +30,7 @@ class Inferencia:
             }
         ]
 
+        logc.gerar_log(mensagem_log="OBTENDO RESPOSTA DA PIPE...")
         output = pipe(text=messages, max_new_tokens=200)
 
         return output[0]["generated_text"][-1]["content"]