Update README.md

README.md

@@ -1,40 +1,23 @@
 ---
-library_name: transformers
-tags: []
+license: mit
+language:
+- en
+base_model:
+- google-t5/t5-base
+datasets:
+- abisee/cnn_dailymail
+metrics:
+- rouge
 ---
+# T5-Base-Sum
 
-# Model Card for Model ID
+This model is a fine-tuned version of `T5` for summarization tasks. It was trained on various articles and is hosted on Hugging Face for easy access and use.
 
-<!-- Provide a quick summary of what the model is/does. -->
+## Model Usage
 
+Below is an example of how to load and use this model for summarization:
 
-
-## Model Details
-
-### Model Description
-
-<!-- Provide a longer summary of what this model is. -->
-
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
-
-- **Developed by:** [More Information Needed]
-- **Funded by [optional]:** [More Information Needed]
-- **Shared by [optional]:** [More Information Needed]
-- **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
-- **Finetuned from model [optional]:** [More Information Needed]
-
-### Model Sources [optional]
-
-<!-- Provide the basic links for the model. -->
-
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
-
-## Uses
-''' Python 
+```python
 import torch
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
@@ -60,7 +43,6 @@ input_prompts = [
 ]
 
 # Generate responses
-
 generated_responses = {}
 for prompt in input_prompts:
     inputs = tokenizer(prompt, return_tensors="pt", max_length=400, truncation=True, padding="max_length").to(device)
@@ -82,8 +64,6 @@ for prompt in input_prompts:
     generated_responses[prompt] = generated_text
 
 # Display the input prompts and the generated responses
-
 for prompt, response in generated_responses.items():
     print(f"Prompt: {prompt}")
     print(f"Response: {response}\n")
-