Fta98 commited on
Commit
1c45748
·
1 Parent(s): 039b8c1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def generate_response(user_query: str, system_prompt: str, messages: str="", tem
66
  repetition_penalty=repetition_penalty,
67
  )
68
  #response = tokenizer.decode(tokens[0][inputs.shape[1]:], skip_special_tokens=True).strip()
69
- return tokens[0][inputs.shape[1]:]
70
 
71
 
72
  st.header(":dna: 遺伝カウンセリング対話AI")
 
66
  repetition_penalty=repetition_penalty,
67
  )
68
  #response = tokenizer.decode(tokens[0][inputs.shape[1]:], skip_special_tokens=True).strip()
69
+ return inputs
70
 
71
 
72
  st.header(":dna: 遺伝カウンセリング対話AI")