Added inference reference
Browse files
README.md
CHANGED
@@ -21,6 +21,26 @@ language:
|
|
21 |
|
22 |
**Uso Recomendado:** O Legislinho pode ser usado por indivíduos que procuram uma compreensão básica de assuntos legais no Brasil. É recomendável que os usuários procurem orientação legal profissional para situações específicas ou complexas.
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
**Avaliação:** O modelo ainda não foi avaliado em um ambiente de produção.
|
25 |
|
26 |
**Contato:** Para mais informações ou dúvidas sobre o Legislinho, entre em contato conosco. [email](costa@maguscorp.com.br)
|
|
|
21 |
|
22 |
**Uso Recomendado:** O Legislinho pode ser usado por indivíduos que procuram uma compreensão básica de assuntos legais no Brasil. É recomendável que os usuários procurem orientação legal profissional para situações específicas ou complexas.
|
23 |
|
24 |
+
**Inferencia:** Para executar inferencia se necessita ter os seguintes pacotes instalados:
|
25 |
+
```
|
26 |
+
(pip install / poetry add)peft accelerate bitsandbytes safetensors transformers
|
27 |
+
```
|
28 |
+
Códico completo para inferência:
|
29 |
+
```
|
30 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
31 |
+
model = AutoModelForCausalLM.from_pretrained('MagusCorp/legislinho')
|
32 |
+
tokenizer = AutoTokenizer.from_pretrained('MagusCorp/legislinho')
|
33 |
+
preprompt = 'Você é um agente de auxílio jurídico que não repete palavras de forma redundante e redige de forma clara concisa, como alguém formado em direito. Você não cita leis de forma alguma apenas procedimentos legais e operações. Você apenas é uma primeira consulta antes da pessoa procurar um advogado. Utilize todo o seu conhecimento da constituição brasileira para responder. Não repita a mesma informação em sua resposta. Seja assertivo. Atente-se a pergunta e elabore uma estratégia de solução do problema contemplando toda a pergunta, tal como faz um advogado. Pergunta:\n'
|
34 |
+
pospromt = "\nResposta:"
|
35 |
+
|
36 |
+
prompt = 'Minha mae vendeu a casa dela sem consultar nem um dos filhos ela pode fazer isso?'
|
37 |
+
|
38 |
+
inputs = tokenizer(preprompt+prompt+pospromt, return_tensors='pt').to('cuda')
|
39 |
+
outputs = model.generate(**inputs,max_new_tokens=1000)
|
40 |
+
answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
41 |
+
print(answer.split(prompt)[1])
|
42 |
+
```
|
43 |
+
|
44 |
**Avaliação:** O modelo ainda não foi avaliado em um ambiente de produção.
|
45 |
|
46 |
**Contato:** Para mais informações ou dúvidas sobre o Legislinho, entre em contato conosco. [email](costa@maguscorp.com.br)
|