Update app.py

app.py

@@ -3,9 +3,11 @@ import gradio as gr
 import transformers
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
 # Load the tokenizer and model
-tokenizer = GPT2Tokenizer.from_pretrained('DarwinAnim8or/gpt-grug-1.5b')
-model = GPT2LMHeadModel.from_pretrained('DarwinAnim8or/gpt-grug-1.5b')
+tokenizer = GPT2Tokenizer.from_pretrained('DarwinAnim8or/gpt-grug-1.5b', use_auth_token=HF_TOKEN)
+model = GPT2LMHeadModel.from_pretrained('DarwinAnim8or/gpt-grug-1.5b', use_auth_token=HF_TOKEN)
 
 # Define the function that generates text from a prompt
 def generate_text(prompt):