artificialguybr commited on
Commit
a85a91b
1 Parent(s): 408fb2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -47,10 +47,12 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
47
 
48
  # Gerar a saída
49
  output = model.generate(input_ids=input_ids, temperature=temperature, do_sample=True, top_p=top_p, top_k=top_k, max_length=max_tokens)
50
-
 
51
  # Decodificar a saída
52
  decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
53
-
 
54
  # Atualizar o histórico
55
  history[-1][1] += decoded_output
56
 
 
47
 
48
  # Gerar a saída
49
  output = model.generate(input_ids=input_ids, temperature=temperature, do_sample=True, top_p=top_p, top_k=top_k, max_length=max_tokens)
50
+ print("Output:")
51
+ print(output)
52
  # Decodificar a saída
53
  decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
54
+ print("Decoded_Output:")
55
+ print(decoded_output)
56
  # Atualizar o histórico
57
  history[-1][1] += decoded_output
58