Hugging Face's logo

Spaces:
Chemsseddine
/
summarisation
Sleeping

App Files Community
Chemsseddine commited on
Commit
2cd241b
·
1 Parent(s): 7db5c99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import RobertaTokenizerFast, EncoderDecoderModel
2
  import torch
3
  #initialisation de tokenizer
4
  device = "cuda" if torch.cuda.is_available() else "cpu"
5
- tokenizer = RobertaTokenizerFast.from_pretrained("Chemsseddine/bert2gpt2SUMM-finetuned-mlsum")
6
  #aider les token special
7
  tokenizer.bos_token = tokenizer.cls_token
8
  tokenizer.eos_token = tokenizer.sep_token
 
2
  import torch
3
  #initialisation de tokenizer
4
  device = "cuda" if torch.cuda.is_available() else "cpu"
5
+ tokenizer = RobertaTokenizerFast.from_pretrained("Chemsseddine/bert2gpt2SUMM-finetuned-mlsum")#Chemsseddine/bert2gpt2SUMM-finetuned-mlsum
6
  #aider les token special
7
  tokenizer.bos_token = tokenizer.cls_token
8
  tokenizer.eos_token = tokenizer.sep_token