Najish Mahmud commited on
Commit
4731905
·
1 Parent(s): 1147014

added formats

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. app.py +4 -2
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("csebuetnlp/banglat5", use_fast=False)
 
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()