fix
Browse files
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 |
-
|
69 |
-
return
|
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")
|