Salvando alterações

Inferencia.py

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

app.py

@@ -30,8 +30,9 @@ if arquivos_upload is not None:
         )
 
         st.write(dados_imagem)
+        print(dados_imagem)
 
-        logc.gerar_log(mensagem_log=f"DEFININDO TIPO DE DOCUMENTO: {dados_imagem}")
+        # logc.gerar_log(mensagem_log=f"DEFININDO TIPO DE DOCUMENTO: {dados_imagem}")
 
         # dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
         #