Update app.py

app.py

@@ -1,11 +1,16 @@
+import os
 import gradio as gr
 import random
 from transformers import pipeline
 
 
+# Get the Hugging Face token from environment variables
+hf_token = os.getenv("gpt2_tokens")
 
+if not hf_token:
+    raise ValueError("Hugging Face token not found. Please set HF_TOKEN as an environment variable.")
 # Load the text generation pipeline with your fine-tuned model
-generator = pipeline('text-generation', model='isitcoding/gpt2_120_finetuned')
+generator = pipeline('text-generation', model='isitcoding/gpt2_120_finetuned', use_auth_token = hf_token)
 
 # Function to generate responses using the text generation model
 def respond(message, chat_history):