Update app.py

app.py

@@ -2,7 +2,8 @@ import os
 import re
 import pdfminer
 from pdfminer.high_level import extract_pages
-from transformers import pipeline, AutoModelForQuestionAnswering, AutoTokenizer
+from transformers import pipeline, TFBertForQuestionAnswering, AutoTokenizer
+
 
 import streamlit as st
 
@@ -41,9 +42,9 @@ def answer_question(text, question):
     Returns:
         The answer extracted from the text using the model.
     """
-    qa_model_name = "deepset/roberta-base-squad2"  # Replace with your chosen model
+    qa_model_name = "bert-base-uncased"  # Replace with your chosen model
 
-    qa_model = AutoModelForQuestionAnswering.from_pretrained(qa_model_name)
+    qa_model = TFBertForQuestionAnswering.from_pretrained(qa_model_name)
     tokenizer = AutoTokenizer.from_pretrained(qa_model_name)
 
     inputs = tokenizer(question, text, return_tensors="pt")  # Tokenize inputs