Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def get_llm():
|
|
64 |
model=model,
|
65 |
tokenizer=tokenizer,
|
66 |
max_new_tokens=550,
|
67 |
-
temperature=0.
|
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 |
}
|