Update README.md

README.md

@@ -54,11 +54,11 @@ sample_dict = ProteinSolubilityTask.data_preprocessing(
     sample_dict=sample_dict,
     protein_sequence_key="protein_seq",
     tokenizer_op=tokenizer_op,
-    device=nn_model.device,
+    device=model.device,
 )
 
 # running in generate mode
-batch_dict = nn_model.generate(
+batch_dict = model.generate(
     [sample_dict],
     output_scores=True,
     return_dict_in_generate=True,
@@ -81,7 +81,7 @@ For more advanced usage, see our detailed example at: on `https://github.com/Bio
 
 ## Citation
 
-If you found our work useful, please consider to give a star to the repo and cite our paper:
+If you found our work useful, please consider giving a star to the repo and cite our paper:
 ```
 @article{TBD,
   title={TBD},