Update README.md

README.md

@@ -9,7 +9,7 @@ inference:
     diversity_penalty: 3.01
     no_repeat_ngram_size: 2
     temperature: 0.8
-    max_length: 128
+    max_length: 64
 widget:
 - text: >-
     Learn to build generative AI applications with an expert AWS instructor with the 2-day Developing Generative AI Applications on AWS course.
@@ -27,7 +27,9 @@ widget:
 
 # Text Rewriter Paraphraser
 
-This repository contains a fine-tuned text-rewriting model based on the T5-Base with 223M parameters. 
+This repository contains a fine-tuned text-rewriting model based on the T5-Base with 223M parameters.
+
+Developed by: https://exnrt.com
 
 ## Key Features:
 
@@ -49,13 +51,12 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # Replace 'YOUR_TOKEN' with your actual Hugging Face access token
 tokenizer = AutoTokenizer.from_pretrained("Ateeqq/Text-Rewriter-Paraphraser", token='YOUR_TOKEN')
 model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/Text-Rewriter-Paraphraser", token='YOUR_TOKEN')
-```
-```python
+
 text = "Data science is a field that deals with extracting knowledge and insights from data. "
 
 inputs = tokenizer(text, return_tensors="pt")
 
-output = model.generate(**inputs, max_length=50)
+output = model.generate(**inputs, max_length=64)
 
 print(tokenizer.decode(output[0]))
 ```