Mr-Vicky-01
/

Bart-Finetuned-conversational-summarization

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

Mr-Vicky-01 commited on Apr 2

Commit

138506a

•

1 Parent(s): 265a664

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -54,8 +54,8 @@ provided a simple snippet of how to use this model for the task of paragraph sum
 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("Mr-Vicky-01/conversational_sumarization")
-model = AutoModelForSeq2SeqLM.from_pretrained("Mr-Vicky-01/conversational_sumarization")
 def generate_summary(text):
     inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)

 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("Mr-Vicky-01/Bart-Finetuned-conversational-summarization")
+model = AutoModelForSeq2SeqLM.from_pretrained("Mr-Vicky-01/Bart-Finetuned-conversational-summarization")
 def generate_summary(text):
     inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)