amn-sdqi commited on
Commit
121365e
·
1 Parent(s): 6dc7ab5

prompt changes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ if st.session_state.context:
47
  if question:
48
  result = qa(question=question, context=st.session_state.context)
49
 
50
- prompt = f"{question}\n{result['answer']}"
51
  generated = text_gen(prompt, max_length=100, do_sample=True)[0]['generated_text']
52
 
53
  # save convo
 
47
  if question:
48
  result = qa(question=question, context=st.session_state.context)
49
 
50
+ prompt = f"{st.session_state.context}\n{question}\n"
51
  generated = text_gen(prompt, max_length=100, do_sample=True)[0]['generated_text']
52
 
53
  # save convo