Spaces:
Sleeping
Sleeping
Fernando Cervan
commited on
Commit
·
807e236
1
Parent(s):
7994749
Salvando alterações
Browse files- Inferencia.py +8 -9
- app.py +2 -1
Inferencia.py
CHANGED
@@ -6,16 +6,15 @@ from LogCustomizado import LogCustomizado as logc
|
|
6 |
class Inferencia:
|
7 |
|
8 |
def __init__(self):
|
9 |
-
|
10 |
-
|
11 |
-
|
|
|
|
|
12 |
"image-text-to-text",
|
13 |
-
model=
|
14 |
torch_dtype=torch.bfloat16 # Otimiza para rodar na CPU
|
15 |
)
|
16 |
-
|
17 |
-
def extrair_dados_imagem(self, prompt, imagem):
|
18 |
-
|
19 |
messages = [
|
20 |
{
|
21 |
"role": "system",
|
@@ -31,8 +30,8 @@ class Inferencia:
|
|
31 |
}
|
32 |
]
|
33 |
|
34 |
-
logc.gerar_log(mensagem_log=
|
35 |
-
output =
|
36 |
|
37 |
return output[0]["generated_text"][-1]["content"]
|
38 |
|
|
|
6 |
class Inferencia:
|
7 |
|
8 |
def __init__(self):
|
9 |
+
...
|
10 |
+
|
11 |
+
def extrair_dados_imagem(self, prompt, imagem):
|
12 |
+
logc.gerar_log(mensagem_log="EXECUTANDO PIPELINE...")
|
13 |
+
pipe = pipeline(
|
14 |
"image-text-to-text",
|
15 |
+
model="google/gemma-3-4b-it",
|
16 |
torch_dtype=torch.bfloat16 # Otimiza para rodar na CPU
|
17 |
)
|
|
|
|
|
|
|
18 |
messages = [
|
19 |
{
|
20 |
"role": "system",
|
|
|
30 |
}
|
31 |
]
|
32 |
|
33 |
+
logc.gerar_log(mensagem_log="OBTENDO RESPOSTA DA PIPE...")
|
34 |
+
output = pipe(text=messages, max_new_tokens=400)
|
35 |
|
36 |
return output[0]["generated_text"][-1]["content"]
|
37 |
|
app.py
CHANGED
@@ -30,8 +30,9 @@ if arquivos_upload is not None:
|
|
30 |
)
|
31 |
|
32 |
st.write(dados_imagem)
|
|
|
33 |
|
34 |
-
logc.gerar_log(mensagem_log=f"DEFININDO TIPO DE DOCUMENTO: {dados_imagem}")
|
35 |
|
36 |
# dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
|
37 |
#
|
|
|
30 |
)
|
31 |
|
32 |
st.write(dados_imagem)
|
33 |
+
print(dados_imagem)
|
34 |
|
35 |
+
# logc.gerar_log(mensagem_log=f"DEFININDO TIPO DE DOCUMENTO: {dados_imagem}")
|
36 |
|
37 |
# dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
|
38 |
#
|