suriya7 commited on
Commit
f17dab6
·
verified ·
1 Parent(s): 05fe813

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -33,9 +33,11 @@ pip install torch
33
  ```
34
  ## Inference
35
  ```python
36
- from transformers import T5ForConditionalGeneration, T5Tokenizer
 
37
 
38
- model_name = "MaxMini-Instruct-248M"
 
39
 
40
  my_question = "what is depression?"
41
  inputs = "Please answer to this question: " + my_question
 
33
  ```
34
  ## Inference
35
  ```python
36
+ # Load model directly
37
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
38
 
39
+ tokenizer = AutoTokenizer.from_pretrained("suriya7/MaxMini-Instruct-248M")
40
+ model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/MaxMini-Instruct-248M")
41
 
42
  my_question = "what is depression?"
43
  inputs = "Please answer to this question: " + my_question