Update model.py

model.py

@@ -25,7 +25,7 @@ from transformers import (
     PreTrainedModel,
     PreTrainedTokenizer
 )
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, login
 from langchain.prompts import PromptTemplate
 
 # Vector database