Salvando alterações

Inferencia.py

@@ -6,15 +6,16 @@ from LogCustomizado import LogCustomizado as logc
 class Inferencia:
 
     def __init__(self):
-        ...
-
-    def extrair_dados_imagem(self, prompt, imagem):
-        logc.gerar_log(mensagem_log="EXECUTANDO PIPELINE...")
-        pipe = pipeline(
+        self.modelo = "google/gemma-3-4b-it"
+        logc.gerar_log(mensagem_log="CARREGANDO MODELO")
+        self.pipe = pipeline(
             "image-text-to-text",
-            model="google/gemma-3-4b-it",
+            model=self.modelo,
             torch_dtype=torch.bfloat16  # Otimiza para rodar na CPU
         )
+
+    def extrair_dados_imagem(self, prompt, imagem):
+
         messages = [
             {
                 "role": "system",
@@ -30,8 +31,8 @@ class Inferencia:
             }
         ]
 
-        logc.gerar_log(mensagem_log="OBTENDO RESPOSTA DA PIPE...")
-        output = pipe(text=messages, max_new_tokens=2000)
+        logc.gerar_log(mensagem_log=f"OBTENDO RESPOSTA DO MODELO {self.modelo}")
+        output = self.pipe(text=messages, max_new_tokens=2000)
 
         return output[0]["generated_text"][-1]["content"]