Update app.py

app.py

@@ -41,12 +41,10 @@ if st.sidebar.button("Submit"):
 
         # Define the prompt format for the model
         prompt = """
-        ### Instruction and Input:
-        Based on the following context/document:
         {}
+        
         Please answer the question: {}
 
-        ### Response:
         {}
         """
 
@@ -69,7 +67,11 @@ if st.sidebar.button("Submit"):
 
         # Decode and display the results
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        st.write(response)
+
+        # Remove unwanted text fields from the response
+        clean_response = response.replace("### Instruction and Input:", "").replace("### Response:", "").strip()
+
+        st.write(clean_response)
     else:
         st.sidebar.error("Please upload a PDF file and enter a query.")