Update app.py

app.py

@@ -1,23 +1,3 @@
-import streamlit as st
-import PyPDF2
-from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
-
-# Load PDF and extract text
-def load_pdf(uploaded_file):
-    reader = PyPDF2.PdfReader(uploaded_file)
-    text = ""
-    for page in reader.pages:
-        text += page.extract_text() + "\n"
-    return text
-
-# Initialize RAG model
-def initialize_rag_model():
-    # Load the tokenizer and model
-    tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")
-    retriever = RagRetriever.from_pretrained("facebook/rag-token-nq", index_name="legacy", use_dummy_dataset=True)
-    model = RagTokenForGeneration.from_pretrained("facebook/rag-token-nq")
-    return tokenizer, retriever, model
-
 # Process user query
 def generate_answer(query, context, tokenizer, retriever, model):
     # Tokenize the input question
@@ -43,4 +23,4 @@ if uploaded_file is not None:
     user_query = st.text_input("Ask a question about the PDF:")
     if user_query:
         answer = generate_answer(user_query, text, tokenizer, retriever, model)
-        st.write("Answer: {answer}"
+        st.write(f"Answer: {answer}")  # Corrected line with closing parenthesis