Hugging Face's logo

Spaces:
llmahmad
/
Summarization
Sleeping

App Files Community
llmahmad commited on
Commit
9758b14
·
verified ·
1 Parent(s): c4f6d26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import streamlit as st
5
  from transformers import BartTokenizer, BartForConditionalGeneration
6
 
7
  # Load the model and tokenizer
8
- model_name = 'llmahmad/facebook_BART_summary'
9
 
10
  tokenizer = BartTokenizer.from_pretrained(model_name)
11
  model = BartForConditionalGeneration.from_pretrained(model_name)
 
5
  from transformers import BartTokenizer, BartForConditionalGeneration
6
 
7
  # Load the model and tokenizer
8
+ model_name = 'facebook/bart-large-cnn'
9
 
10
  tokenizer = BartTokenizer.from_pretrained(model_name)
11
  model = BartForConditionalGeneration.from_pretrained(model_name)