Update app.py

app.py

@@ -4,15 +4,20 @@ import gradio as gr
 from text_generation import Client  # Assuming you have a text_generation module
 from transformers import AutoModel
 
-# Set Hugging Face API token
-os.environ['HF_READ_TOKEN'] = "your_actual_token_here"  # Replace with your Hugging Face API token
-
+# Set Hugging Face API token from environment variable
 HF_TOKEN = os.environ.get('HF_READ_TOKEN', False)
+
+if not HF_TOKEN:
+    raise ValueError("Hugging Face API token is not set. Set the HF_READ_TOKEN environment variable.")
+
 EOS_STRING = '</s>'
 EOT_STRING = '<EOT>'
 
 # Load the private model with access token
-access_token = "hf_..."  # Replace with your actual access token
+access_token = os.environ.get('HF_MODEL_ACCESS_TOKEN', False)
+if not access_token:
+    raise ValueError("Hugging Face model access token is not set. Set the HF_MODEL_ACCESS_TOKEN environment variable.")
+
 model = AutoModel.from_pretrained("private/model", token=access_token)
 
 def get_prompt(message, chat_history, system_prompt):