Hugging Face's logo

Spaces:
JDmayo
/
mbart
Build error

App Files Community
JDmayo commited on
Commit
2fa14b5
·
verified ·
1 Parent(s): 18037bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  from transformers import BartTokenizer, BartForConditionalGeneration
6
 
7
  # Replace with your Hugging Face model repository path
8
- model_repo_path = 'AbdurRehman313/BART_samsum'
9
 
10
  # Load the model and tokenizer
11
  model = BartForConditionalGeneration.from_pretrained(model_repo_path)
 
5
  from transformers import BartTokenizer, BartForConditionalGeneration
6
 
7
  # Replace with your Hugging Face model repository path
8
+ model_repo_path = 'JDmayo/mbart-base'
9
 
10
  # Load the model and tokenizer
11
  model = BartForConditionalGeneration.from_pretrained(model_repo_path)