aaliyaan commited on
Commit
1775594
·
1 Parent(s): ddf2105
Files changed (1) hide show
  1. app.py +2 -4
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", use_fast=False),
15
  },
16
  "Broken Answer (T0pp)": {
17
  "model": AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp"),
18
- "tokenizer": AutoTokenizer.from_pretrained("bigscience/T0pp", use_fast=False),
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