Update README.md

README.md

@@ -41,7 +41,7 @@ pip install transformers
 Uncontrollable generation can be handled via prompting the model with the phrase 'Seq=<'. 
 
 ```
-generator = pipeline('text-generation', model="Esperanto/Protein-Phi-3-mini", trust_remote_code=True)
+generator = pipeline('text-generation', model="Esperanto/Protein-Phi-3-mini", tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
 
 sequences = generator("Seq=<",temperature=0.2,
     top_k=40,
@@ -61,7 +61,7 @@ for sequence in sequences:
 Controllable generation can be done by prompting the model with '[Generate xxx protein] Seq=<'. Here, xxx can be any family from the 10 classes supported by this model.
 
 ```
-generator = pipeline('text-generation', model="Esperanto/Protein-Phi-3-mini", trust_remote_code=True)
+generator = pipeline('text-generation', model="Esperanto/Protein-Phi-3-mini", tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
 
 sequences = generator("[Generate Ligase enzyme protein] Seq=<",temperature=0.2,
     top_k=40,