Update app.py

app.py

@@ -47,7 +47,7 @@ if prompt := st.chat_input("Say Something", key="input_1", disabled=st.session_s
 if prompt or st.session_state.state_chat_input:
     if st.session_state.state_chat_input:
         with st.spinner(text="Generating response..."):
-            model_inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+            model_inputs = tokenizer(st.session_state.messages[-1]['content'], return_tensors="pt").to(model.device)
             print(model_inputs)
             generated_ids = model.generate(
                 **model_inputs,