Spaces:

xTorch8
/

bart-summarizer

Sleeping

xTorch8 commited on May 21

Commit

5d23333

1 Parent(s): ec8f198

Fix token

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ TOKEN = os.getenv("TOKEN")
 MAX_TOKENS = 1024
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL, token = TOKEN)
-tokenizer = AutoTokenizer.from_pretrained(MODEL, TOKEN)
 def summarize_text(text):
     chunk_size = MAX_TOKENS * 4

 MAX_TOKENS = 1024
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL, token = TOKEN)
+tokenizer = AutoTokenizer.from_pretrained(MODEL, token = TOKEN)
 def summarize_text(text):
     chunk_size = MAX_TOKENS * 4