woman
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
|
3 |
from PyPDF2 import PdfReader
|
4 |
|
5 |
-
|
6 |
# Models and Tokenizers Setup
|
7 |
models = {
|
8 |
"Text Generator (Bloom)": {
|
@@ -11,11 +10,11 @@ models = {
|
|
11 |
},
|
12 |
"PDF Summarizer (T5)": {
|
13 |
"model": AutoModelForSeq2SeqLM.from_pretrained("aaliyaan/t5-small-finetuned-career"),
|
14 |
-
"tokenizer": AutoTokenizer.from_pretrained("aaliyaan/t5-small-finetuned-career"
|
15 |
},
|
16 |
"Broken Answer (T0pp)": {
|
17 |
"model": AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp"),
|
18 |
-
"tokenizer": AutoTokenizer.from_pretrained("bigscience/T0pp"
|
19 |
},
|
20 |
}
|
21 |
|
@@ -54,7 +53,6 @@ def chat_with_model(model_choice, user_message, chat_history, file=None):
|
|
54 |
|
55 |
# Function to Extract Text from PDF
|
56 |
def extract_text_from_pdf(file):
|
57 |
-
from PyPDF2 import PdfReader
|
58 |
reader = PdfReader(file.name)
|
59 |
text = "\n".join(page.extract_text() for page in reader.pages if page.extract_text())
|
60 |
return text
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
|
3 |
from PyPDF2 import PdfReader
|
4 |
|
|
|
5 |
# Models and Tokenizers Setup
|
6 |
models = {
|
7 |
"Text Generator (Bloom)": {
|
|
|
10 |
},
|
11 |
"PDF Summarizer (T5)": {
|
12 |
"model": AutoModelForSeq2SeqLM.from_pretrained("aaliyaan/t5-small-finetuned-career"),
|
13 |
+
"tokenizer": AutoTokenizer.from_pretrained("aaliyaan/t5-small-finetuned-career"),
|
14 |
},
|
15 |
"Broken Answer (T0pp)": {
|
16 |
"model": AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp"),
|
17 |
+
"tokenizer": AutoTokenizer.from_pretrained("bigscience/T0pp"),
|
18 |
},
|
19 |
}
|
20 |
|
|
|
53 |
|
54 |
# Function to Extract Text from PDF
|
55 |
def extract_text_from_pdf(file):
|
|
|
56 |
reader = PdfReader(file.name)
|
57 |
text = "\n".join(page.extract_text() for page in reader.pages if page.extract_text())
|
58 |
return text
|