samuel-moreira commited on
Commit
1c3a7a2
1 Parent(s): 5d0f8a4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -18,6 +18,7 @@ app = FastAPI()
18
 
19
  @app.post("/llm_on_cpu")
20
  async def stream(item: validation):
 
21
  system_prompt = """Leia o currículo e classifique as entidades no meu estilo.
22
  Respeite o formato de saída JSON.
23
  Retorne APENAS O DICIONÁRIO JSON e nada mais.
@@ -27,6 +28,5 @@ Geralmente a descrição da experiência vem descrita logo após o cargo. Procur
27
  Acrescente todas as informações que encontrar, todas devem ser classificadas nos seus respectivos campos. Não deixe passar nada.
28
  Caso não encontre alguma informação, retorne o campo com Null."""
29
  E_INST = "</s>"
30
- user = "<|user|>"
31
- prompt = f"{system_prompt}{E_INST}\n{user}\n{item.prompt}{E_INST}"
32
  return llm(prompt)
 
18
 
19
  @app.post("/llm_on_cpu")
20
  async def stream(item: validation):
21
+ print("Start")
22
  system_prompt = """Leia o currículo e classifique as entidades no meu estilo.
23
  Respeite o formato de saída JSON.
24
  Retorne APENAS O DICIONÁRIO JSON e nada mais.
 
28
  Acrescente todas as informações que encontrar, todas devem ser classificadas nos seus respectivos campos. Não deixe passar nada.
29
  Caso não encontre alguma informação, retorne o campo com Null."""
30
  E_INST = "</s>"
31
+ prompt = f"{system_prompt}{E_INST}\n{item.prompt}{E_INST}"
 
32
  return llm(prompt)