Update app.py

app.py

@@ -73,7 +73,7 @@ if submit_button and query:
     if uploaded_files:
         pdf_text = extract_text_from_pdfs(uploaded_files)
         
-        # Prepare the input prompt without unnecessary headers
+        # Prepare the input prompt
         prompt = f"""
         Based on the following context/document:
         {pdf_text}
@@ -94,12 +94,12 @@ if submit_button and query:
             no_repeat_ngram_size=5,
         )
 
-        # Decode and clean the response
+        # Decode the response and clean it
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        response_clean = response.replace("### Instruction and Input:", "").replace("### Response:", "").strip()
+        clean_response = response.strip()
 
         # Update chat history
-        st.session_state.chat_history.append((query, response_clean))
+        st.session_state.chat_history.append((query, clean_response))
 
 # Display chat history
 if st.session_state.chat_history: