Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,9 +6,10 @@ import os
|
|
6 |
import gradio as gr
|
7 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
8 |
|
|
|
9 |
# Load model directly
|
10 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
11 |
-
model = T5ForConditionalGeneration.from_pretrained(
|
12 |
|
13 |
def summarize_text(input_text):
|
14 |
summar_input = 'summarize: '+input_text
|
|
|
6 |
import gradio as gr
|
7 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
8 |
|
9 |
+
model_name = 'Abijith/Billsum-text-summarizer-t5-small'
|
10 |
# Load model directly
|
11 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
12 |
+
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
13 |
|
14 |
def summarize_text(input_text):
|
15 |
summar_input = 'summarize: '+input_text
|