Update README.md
Browse files
README.md
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
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.
|
4 |
|
5 |
## Model Details
|
6 |
-
- **Model Architecture**:
|
7 |
- **Task**: Question Answering (QA-Chatbot)
|
8 |
- **Dataset**: [SQuAD]
|
9 |
- **Quantization**: FP16
|
@@ -25,7 +25,7 @@ import torch
|
|
25 |
|
26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
27 |
|
28 |
-
model_name = "t5-
|
29 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
30 |
model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
|
31 |
```
|
|
|
3 |
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.
|
4 |
|
5 |
## Model Details
|
6 |
+
- **Model Architecture**:t5-qa-chatbot
|
7 |
- **Task**: Question Answering (QA-Chatbot)
|
8 |
- **Dataset**: [SQuAD]
|
9 |
- **Quantization**: FP16
|
|
|
25 |
|
26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
27 |
|
28 |
+
model_name = "AventIQ-AI/t5-qa-chatbot"
|
29 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
30 |
model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
|
31 |
```
|