Spaces:

jkollu
/

bart-multinews-space

Build error

jkollu commited on Apr 10, 2025

Commit

d8decee

verified ·

1 Parent(s): af3ed03

Upload folder using huggingface_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@
 import gradio as gr
 from transformers import BartForConditionalGeneration, BartTokenizer
-model = BartForConditionalGeneration.from_pretrained("your-username/bart-multinews-summarizer")
-tokenizer = BartTokenizer.from_pretrained("your-username/bart-multinews-summarizer")
 def summarize(text):
     inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)

 import gradio as gr
 from transformers import BartForConditionalGeneration, BartTokenizer
+model = BartForConditionalGeneration.from_pretrained("jkollu/multinews-summarizer")
+tokenizer = BartTokenizer.from_pretrained("jkollu/multinews-summarizer")
 def summarize(text):
     inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)