Views
No views yet

1from transformers import AutoTokenizer, pipeline
2import torch
3
4model_id = "dominguesm/Canarim-7B-Instruct"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7pipe = pipeline(
8 "text-generation",
9 model=model_id,
10 torch_dtype=torch.float16,
11 device_map="auto",
12)
13
14
15def make_prompt(instruction, input=None):
16 if input:
17 return f"""Abaixo está uma instrução que descreve uma tarefa, emparelhada com uma entrada que fornece mais contexto. Escreva uma resposta que conclua adequadamente a solicitação.
18
19### Instruções:
20{instruction}
21
22### Entrada:
23{input}
24
25### Resposta:"""
26 else:
27 return f"""Abaixo está uma instrução que descreve uma tarefa. Escreva uma resposta que conclua adequadamente a solicitação.
28
29### Instruções:
30{instruction}
31
32### Resposta:"""
33
34instruction = "Faça um resumo conciso do seguinte texto."
35context = """
36No início da obra, o narrador Bentinho explica como escolheu o título para o seu livro. Assim, ele conta que conheceu um poeta em uma viagem de trem. O jovem recitou alguns versos, mas Bentinho, já velho, fechou os olhos algumas vezes. O poeta, então, achou que ele estava cochilando e, ofendido, lhe chamou de Dom Casmurro.
37
38Em seguida, o narrador inicia a narrativa de suas memórias, que começa quando ele tem 15 anos. Nessa época, Capitu, uma jovem de 14 anos, é sua vizinha. Os adolescentes têm uma forte amizade, o que incomoda o agregado da família, o amante de superlativos José Dias.
39
40Desconfiado da relação entre os jovens, ele sugere à D. Glória, mãe de Bentinho, que já é hora de enviar o rapaz para o seminário. Isso porque, quando o menino nasceu, sua mãe fez uma promessa de que ele seria padre. Dessa forma, se apresenta um obstáculo ao amor do jovem casal, o que, enganosamente, faz parecer que o livro é um romance romântico.
41"""
42
43prompt = make_prompt(question)
44
45sequences = pipe(
46 prompt,
47 do_sample=True,
48 num_return_sequences=1,
49 eos_token_id=tokenizer.eos_token_id,
50 max_length=2048,
51 temperature=0.9,
52 top_p=0.6,
53 repetition_penalty=1.15
54)O trabalho retrata as memórias de Bentinho, um jovem de 15 anos. Ele conta
sobre sua infância na casa de seus pais, onde vive com sua mãe, a dona
Glória, e sua governanta, a governanta Josefa. No entanto, no outono de
1860, sua vida mudará drasticamente. Durante essa temporada, ele conhece
Capitu, uma moça de 14 anos, que se tornará sua companheira por muitos anos.| Metric | Value |
|---|---|
| Average | 47.21 |
| ENEM Challenge (No Images) | 27.50 |
| BLUEX (No Images) | 26.15 |
| OAB Exams | 29.93 |
| Assin2 RTE | 75.74 |
| Assin2 STS | 12.08 |
| FaQuAD NLI | 43.92 |
| HateBR Binary | 79.57 |
| PT Hate Speech Binary | 64.01 |
| tweetSentBR | 66 |