auth

app.py

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

requirements.txt

@@ -1,4 +1,5 @@
 peft 
 accelerate 
 transformers
-gradio
+gradio
+python-dotenv