mdacosta commited on
Commit
2ab4718
1 Parent(s): 4ada41e

Added inference reference

Browse files
Files changed (1) hide show
  1. README.md +20 -0
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)