Fta98 commited on
Commit
039b8c1
·
1 Parent(s): c48b961
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]
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 tokens[0][inputs.shape[1]:]
70
 
71
 
72
  st.header(":dna: 遺伝カウンセリング対話AI")