varshamishra commited on
Commit
3f37597
·
verified ·
1 Parent(s): 37c4a3d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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**:AventIQ-AI/t5-qa-chatbot
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-small"
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
  ```