Spaces:
Runtime error
Runtime error
Commit
·
afeccd2
1
Parent(s):
7a3e1ef
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ import gradio as gr
|
|
| 20 |
|
| 21 |
|
| 22 |
device = 'cuda' if cuda.is_available() else 'cpu'
|
| 23 |
-
tokenizer = AutoTokenizer.from_pretrained("t5-
|
| 24 |
os.makedirs('./transcripts/')
|
| 25 |
|
| 26 |
def clean_text(link,start,end):
|
|
@@ -182,9 +182,8 @@ def clean_text(link,start,end):
|
|
| 182 |
|
| 183 |
def t5_summarizer(link,start, end):
|
| 184 |
input_text = clean_text(link,start,end)
|
| 185 |
-
tokenizer1 = AutoTokenizer.from_pretrained("CareerNinja/t5-large_3e-4")
|
| 186 |
model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5-large_3e-4")
|
| 187 |
-
summarizer1 = pipeline("summarization", model=model1, tokenizer=
|
| 188 |
print(f""" Entered summarizer ! """)
|
| 189 |
out = []
|
| 190 |
for i in range(len(input_text)):
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
device = 'cuda' if cuda.is_available() else 'cpu'
|
| 23 |
+
tokenizer = AutoTokenizer.from_pretrained("CareerNinja/t5-large_3e-4")
|
| 24 |
os.makedirs('./transcripts/')
|
| 25 |
|
| 26 |
def clean_text(link,start,end):
|
|
|
|
| 182 |
|
| 183 |
def t5_summarizer(link,start, end):
|
| 184 |
input_text = clean_text(link,start,end)
|
|
|
|
| 185 |
model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5-large_3e-4")
|
| 186 |
+
summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer)
|
| 187 |
print(f""" Entered summarizer ! """)
|
| 188 |
out = []
|
| 189 |
for i in range(len(input_text)):
|