Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForQuest
 from sentence_transformers import SentenceTransformer
 import fitz  # PyMuPDF
 import os
+import torch  # Import torch module
 
 # Load the models
 summarization_model_name = 'facebook/bart-large-cnn'
@@ -35,7 +36,7 @@ def get_answer(question, context):
     answer_end = torch.argmax(end_positions) + 1
     answer = qa_tokenizer.convert_tokens_to_string(qa_tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end]))
     return answer
-    
+
 # Streamlit app
 st.title("PDF Summarizer and Q&A")
 st.write("Upload a PDF file to get a summary and ask questions about the content.")