Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,9 @@ from transformers import pipeline
|
|
9 |
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
10 |
qa_model = pipeline("question-answering", model="deepset/bert-large-uncased-whole-word-masking-squad2")
|
11 |
|
|
|
|
|
|
|
12 |
def extract_text_from_pdf(pdf_file):
|
13 |
with fitz.open(pdf_file) as pdf:
|
14 |
text = ""
|
|
|
9 |
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
10 |
qa_model = pipeline("question-answering", model="deepset/bert-large-uncased-whole-word-masking-squad2")
|
11 |
|
12 |
+
os.environ["HUGGINGFACE_HUB_TOKEN"] = "ctp-hw"
|
13 |
+
my_key = os.environ["HUGGINGFACE_HUB_TOKEN"]
|
14 |
+
|
15 |
def extract_text_from_pdf(pdf_file):
|
16 |
with fitz.open(pdf_file) as pdf:
|
17 |
text = ""
|