Update README.md

README.md

@@ -12,16 +12,21 @@ inference:
     max_length: 256
 widget:
 - text: >-
-    description: Men Business Wrist Watch Quartz Casual Belt Men's Watch Brown Watch
+    description: Men Business Wrist Watch Quartz Casual Belt Men's Watch Brown
+    Watch
   example_title: Example 1
-- text: >-
-    description: SAMSUNG 16" Galaxy Book4 Pro Laptop PC Computer, Intel Core 7
+- text: 'description: SAMSUNG Galaxy Book-4 Pro Laptop PC Computer, Intel Core 7'
   example_title: Example 2
 - text: >-
-    description: Basics 18-Piece Kitchen Dinnerware Set, Plates, Dishes, Bowls - White
+    description: Basics 18-Piece Kitchen Dinnerware Set, Plates, Dishes, Bowls -
+    White
   example_title: Example 3
 datasets:
 - Ateeqq/Amazon-Product-Description
+language:
+- en
+library_name: transformers
+pipeline_tag: text2text-generation
 ---
 # Product Description Generator
 
@@ -36,25 +41,6 @@ T5 model expects a task related prefix: since it is a description generation tas
 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-tokenizer = AutoTokenizer.from_pretrained("Ateeqq/product-description-generator", token="your_token")
-model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/product-description-generator", token="your_token")
-
-input_text = "18-Piece Kitchen Dinnerware white Set, Plates, Dishes, Bowls"
-
-inputs = tokenizer.encode("description: " + input_text, return_tensors="pt", max_length=128, truncation=True)
-
-outputs = model.generate(inputs, max_length=400, num_beams=2, num_beam_groups=2, num_return_sequences=2, repetition_penalty=1.0, diversity_penalty=3.0, no_repeat_ngram_size=2, temperature=0.9, early_stopping=True)
-
-description = tokenizer.decode(outputs[1], skip_special_tokens=True)
-
-print(description)
-```
-
-## Getting Multiple Outputs
-
-```python
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-
 device = "cuda"
 tokenizer = AutoTokenizer.from_pretrained("Ateeqq/product-description-generator", token='your_token')
 model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/product-description-generator", token='your_token').to(device)
@@ -83,7 +69,7 @@ generate_description(title)
  'The 18 Piece Kitchen Dinnerware Set features a white plate, dish and bowl. This set is made of durable stainless steel with an elegant design that will add elegance to your kitchen.',
  'This 18-piece dinnerware set is made of durable stainless steel. It features a white finish and comes with an easy to clean handle for ease of cleaning. The bowls are dishwasher safe, microwave safe or can be used as tablecloths in the kitchen.',
  'Kitchen Dinnerware is a great addition to your kitchen. This 18-piece set includes four plates, two dishes and three bowls for serving food or beverages in the dining room with an elegant design that will add sophistication to any tabletop setting.',
- "p>This 18-piece dinnerware set is made of high quality stainless steel. The white plate and dish are dishwasher safe, easy to clean with the included lids for ease of use. This dining table set features an elegant design that will add elegance style in your kitchen or living room. It's also perfect for serving food on any occasion like birthday parties, house warming ceremonies, Thanksgiving celebrations etc."]
+ "18-Piece Kitchen Dinnerware White Set, Plates & DisheSize(LxW) Size (HXD), Material is made of high quality stainless steel and the bowl has an elegant design. The dish set includes 1 plate with 2 plates; 3 dishes for 4 cups or 6 serving spoon/bowl 2. This dinner table sets are perfect to serve your food in style while also being easy on clean up 3. These white kitchen dining tables have been designed by professional chefs using only premium materials that will not scratch surfaces 4. Easy cleaning dishwasher safe 5. Durable material can be used as disposable wash cloth 6. Perfect gift idea For family members who love cooking."]
 ```
 
 ## Features