michailroussos commited on
Commit
bafd5e5
·
1 Parent(s): f7f950e

small change

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -50,12 +50,14 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
50
  top_p=top_p,
51
  pad_token_id=tokenizer.eos_token_id, # Ensure padding is replaced with EOS
52
  )
53
-
 
54
  # Decode the generated output
55
  response = tokenizer.decode(
56
  output[0], skip_special_tokens=True
57
  ).strip() # Remove any extra whitespace or unexpected tokens
58
 
 
59
  # Yield the clean response for display
60
  yield response
61
 
 
50
  top_p=top_p,
51
  pad_token_id=tokenizer.eos_token_id, # Ensure padding is replaced with EOS
52
  )
53
+ print("output")
54
+ print(output)
55
  # Decode the generated output
56
  response = tokenizer.decode(
57
  output[0], skip_special_tokens=True
58
  ).strip() # Remove any extra whitespace or unexpected tokens
59
 
60
+
61
  # Yield the clean response for display
62
  yield response
63