Salvando alterações

Inferencia.py

@@ -31,7 +31,7 @@ class Inferencia:
         ]
 
         logc.gerar_log(mensagem_log="OBTENDO RESPOSTA DA PIPE...")
-        output = pipe(text=messages, max_new_tokens=200)
+        output = pipe(text=messages, max_new_tokens=2000)
 
         return output[0]["generated_text"][-1]["content"]
 

app.py

@@ -35,13 +35,13 @@ if arquivos_upload is not None:
 
         dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
 
-        # if dados_imagem_dict["tipo_documento"] == "CNH":
-        #     dados_cnh = inferencia.extrair_dados_imagem(
-        #         PromptDocumento.cnh(),
-        #         imagem_convertida
-        #     )
-        #     logc.gerar_log(mensagem_log=f"DADOS CNH: {dados_cnh}")
-        #     logc.gerar_log(mensagem_log=f"ARQUIVO PROCESSADO: {arquivo.name}")
+        if dados_imagem_dict["tipo_documento"] == "CNH":
+            dados_cnh = inferencia.extrair_dados_imagem(
+                PromptDocumento.cnh(),
+                imagem_convertida
+            )
+            logc.gerar_log(mensagem_log=f"DADOS CNH: {dados_cnh}")
+            logc.gerar_log(mensagem_log=f"ARQUIVO PROCESSADO: {arquivo.name}")
 
     logc.gerar_log(mensagem_log=f"FIM DO PROCESSAMENTO")