AventIQ-AI
/

t5-qa-chatbot

Model card Files Files and versions Community

varshamishra commited on Feb 24

Commit

3f37597

·

verified ·

1 Parent(s): 37c4a3d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -3,7 +3,7 @@
 This repository hosts a fine-tuned version of the **T5-Base** model optimized for question-answering tasks using the [SQuAD] dataset. The model is designed to efficiently perform question answering while maintaining high accuracy.
 ## Model Details
-- **Model Architecture**:AventIQ-AI/t5-qa-chatbot
 - **Task**: Question Answering (QA-Chatbot)
 - **Dataset**: [SQuAD]
 - **Quantization**: FP16
@@ -25,7 +25,7 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_name = "t5-small"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
 ```

 This repository hosts a fine-tuned version of the **T5-Base** model optimized for question-answering tasks using the [SQuAD] dataset. The model is designed to efficiently perform question answering while maintaining high accuracy.
 ## Model Details
+- **Model Architecture**:t5-qa-chatbot
 - **Task**: Question Answering (QA-Chatbot)
 - **Dataset**: [SQuAD]
 - **Quantization**: FP16
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_name = "AventIQ-AI/t5-qa-chatbot"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
 ```