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