tanyuzhou commited on
Commit
4cb7b09
1 Parent(s): caad9d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,12 +48,12 @@ def respond(
48
  top_p=top_p,
49
  )
50
 
 
 
51
  # Decode the generated response
52
  response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
53
  response = response.split(f"老师\n{message}\n星野\n")[1]
54
 
55
- print("response: ", tokenizer.decode(generate_ids[0], skip_special_tokens=True))
56
-
57
  return response
58
 
59
  # Default prompt for the chatbot
 
48
  top_p=top_p,
49
  )
50
 
51
+ print("response: ", tokenizer.decode(generate_ids[0], skip_special_tokens=True))
52
+
53
  # Decode the generated response
54
  response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
55
  response = response.split(f"老师\n{message}\n星野\n")[1]
56
 
 
 
57
  return response
58
 
59
  # Default prompt for the chatbot