k3ybladewielder commited on
Commit
3cad606
·
verified ·
1 Parent(s): 993f244

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -64,7 +64,7 @@ def get_llm():
64
  model=model,
65
  tokenizer=tokenizer,
66
  max_new_tokens=550,
67
- temperature=0.1,
68
  eos_token_id=tokenizer.eos_token_id,
69
  return_full_text=False
70
  )
@@ -159,8 +159,8 @@ def keyword_router(input_text: str) -> str:
159
 
160
  def keyword_router(input_text: str) -> str:
161
  keywords_map = {
162
- "MAQUININHA": ["maquininha", "máquina", "POS", "pagamento físico"],
163
- "COBRANCA_ONLINE": ["link de pagamento", "cobrança online", "pagamento online", "checkout"],
164
  "PDV_ECOMMERCE": ["PDV", "ecommerce", "venda online", "loja virtual"],
165
  "CONTA_DIGITAL": ["conta digital", "pix", "boleto", "transferência", "cartão"]
166
  }
 
64
  model=model,
65
  tokenizer=tokenizer,
66
  max_new_tokens=550,
67
+ temperature=0.6,
68
  eos_token_id=tokenizer.eos_token_id,
69
  return_full_text=False
70
  )
 
159
 
160
  def keyword_router(input_text: str) -> str:
161
  keywords_map = {
162
+ "MAQUININHA": ["maquininha", "máquina", "POS", "pagamento físico", "taxa", "%"],
163
+ "COBRANCA_ONLINE": ["pagamento", "link de pagamento", "cobrança online", "pagamento online", "checkout"],
164
  "PDV_ECOMMERCE": ["PDV", "ecommerce", "venda online", "loja virtual"],
165
  "CONTA_DIGITAL": ["conta digital", "pix", "boleto", "transferência", "cartão"]
166
  }