Spaces:
Runtime error
Runtime error
Najish Mahmud
commited on
Commit
·
4731905
1
Parent(s):
1147014
added formats
Browse files
README.md
CHANGED
@@ -7,6 +7,7 @@ sdk: gradio
|
|
7 |
sdk_version: 3.29.0
|
8 |
app_file: app.py
|
9 |
pinned: false
|
|
|
10 |
---
|
11 |
|
12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
7 |
sdk_version: 3.29.0
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
+
python_version: 3.11.3
|
11 |
---
|
12 |
|
13 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
CHANGED
@@ -3,9 +3,11 @@ import gradio as gr
|
|
3 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
4 |
from normalizer import normalize
|
5 |
|
|
|
6 |
def summarize(text):
|
7 |
model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5")
|
8 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
|
|
9 |
|
10 |
input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
|
11 |
generated_tokens = model.generate(input_ids)
|
@@ -15,4 +17,4 @@ def summarize(text):
|
|
15 |
|
16 |
gr_interface = gr.Interface("models/Afsara/fb_bart_large_cnn")
|
17 |
|
18 |
-
gr_interface.launch()
|
|
|
3 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
4 |
from normalizer import normalize
|
5 |
|
6 |
+
|
7 |
def summarize(text):
|
8 |
model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5")
|
9 |
+
tokenizer = AutoTokenizer.from_pretrained(
|
10 |
+
"csebuetnlp/banglat5", use_fast=False)
|
11 |
|
12 |
input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
|
13 |
generated_tokens = model.generate(input_ids)
|
|
|
17 |
|
18 |
gr_interface = gr.Interface("models/Afsara/fb_bart_large_cnn")
|
19 |
|
20 |
+
gr_interface.launch()
|