rhaymison commited on
Commit
03caa35
1 Parent(s): 98ca8e1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -211,14 +211,14 @@ pipe = pipeline("text-generation",
211
  )
212
 
213
 
214
- def format_dataset(question:str):
215
  system_prompt = "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido."
216
 
217
  return f"""<|begin_of_text|><|start_header_id|>system<|end_header_id|>
218
  { system_prompt }<|eot_id|><|start_header_id|>user<|end_header_id|>
219
  { question }<|eot_id|><|start_header_id|>assistant<|end_header_id|>"""
220
 
221
- prompt = format_dataset("Me explique quem eram os Romanos")
222
  result = pipe(prompt)
223
  result[0]["generated_text"].split("assistant<|end_header_id|>")[1]
224
 
 
211
  )
212
 
213
 
214
+ def format_prompt(question:str):
215
  system_prompt = "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido."
216
 
217
  return f"""<|begin_of_text|><|start_header_id|>system<|end_header_id|>
218
  { system_prompt }<|eot_id|><|start_header_id|>user<|end_header_id|>
219
  { question }<|eot_id|><|start_header_id|>assistant<|end_header_id|>"""
220
 
221
+ prompt = format_prompt("Me explique quem eram os Romanos")
222
  result = pipe(prompt)
223
  result[0]["generated_text"].split("assistant<|end_header_id|>")[1]
224