ozcangundes commited on
Commit
f73e752
•
1 Parent(s): 171dcb6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -18,11 +18,11 @@ excluding any supervised training. Therefore, the mT5 model has to be fine-tuned
18
  ## Usage 🚀
19
 
20
  ```python
21
- import torch
22
- from transformers import MT5ForConditionalGeneration, MT5Tokenizer
23
 
24
- tokenizer = MT5Tokenizer.from_pretrained("ozcangundes/mt5-small-turkish-squad")
25
- model = MT5ForConditionalGeneration.from_pretrained("ozcangundes/mt5-small-turkish-squad")
 
26
 
27
  def get_answer(question,context):
28
  source_encoding=tokenizer(
 
18
  ## Usage 🚀
19
 
20
  ```python
21
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
22
 
23
+ tokenizer = AutoTokenizer.from_pretrained("ozcangundes/mt5-small-turkish-squad")
24
+
25
+ model = AutoModelForSeq2SeqLM.from_pretrained("ozcangundes/mt5-small-turkish-squad")
26
 
27
  def get_answer(question,context):
28
  source_encoding=tokenizer(