Update app.py

app.py

@@ -11,7 +11,8 @@ from transformers import AutoConfig
 import torch.nn.functional as F
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import UploadFile, File
-import fitz  
+import fitz
+from transformers import AutoModelForSequenceClassification, AutoModelForQuestionAnswering
 
 
 pdf_cache = {"text": None}
@@ -43,7 +44,9 @@ def anq():
     
     qnap = "krrishsinha/nlpques-ans"
     
-    k = pipeline("question-answering", model= qnap, tokenizer=qnap)
+    tokenizer = AutoTokenizer.from_pretrained(qnap, use_fast=False)
+    model = AutoModelForQuestionAnswering.from_pretrained(qnap)
+    k = pipeline("question-answering", model=model, tokenizer=tokenizer)
     
     return k