Update app.py

app.py

@@ -1,5 +1,6 @@
 from peft import PeftModel, PeftConfig
-from transformers import AutoModelForCausalLM, AutoTokenizer, whoami
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
 import gradio as gr
 from dotenv import load_dotenv
 import os
@@ -7,8 +8,9 @@ import os
 
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
+login()
 
-whoami(token=HF_TOKEN)
+# whoami(token=HF_TOKEN)
 
 config = PeftConfig.from_pretrained("pranjalpandey/gemma-open-instruct")
 model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")