Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ with input_container:
|
|
| 70 |
if user_input:
|
| 71 |
with st.spinner("Generating response..."):
|
| 72 |
try:
|
| 73 |
-
response = st.session_state["qa_pipeline"](user_input, max_length=300)
|
| 74 |
generated_text = response[0]["generated_text"]
|
| 75 |
st.session_state["conversation"].append(("You", user_input))
|
| 76 |
st.session_state["conversation"].append(("Model", generated_text))
|
|
|
|
| 70 |
if user_input:
|
| 71 |
with st.spinner("Generating response..."):
|
| 72 |
try:
|
| 73 |
+
response = st.session_state["qa_pipeline"](f"Q: {user_input", max_length=300)
|
| 74 |
generated_text = response[0]["generated_text"]
|
| 75 |
st.session_state["conversation"].append(("You", user_input))
|
| 76 |
st.session_state["conversation"].append(("Model", generated_text))
|