Update README.md
Browse files
README.md
CHANGED
@@ -211,14 +211,14 @@ pipe = pipeline("text-generation",
|
|
211 |
)
|
212 |
|
213 |
|
214 |
-
def
|
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 =
|
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 |
|