Update app.py

app.py

@@ -5,7 +5,7 @@ from text_generation import Client  # Assuming you have a text_generation module
 from transformers import AutoModel
 
 # Set Hugging Face API token from environment variable
-HF_TOKEN = os.environ.get('HF_READ_TOKEN', False)
+HF_TOKEN = os.environ.get('chatbot', False)
 
 if not HF_TOKEN:
     raise ValueError("Hugging Face API token is not set. Set the HF_READ_TOKEN environment variable.")
@@ -18,6 +18,10 @@ 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.")
 
+# Set protected namespaces to resolve the warning
+model_config = AutoModel.config
+model_config['protected_namespaces'] = ()
+
 model = AutoModel.from_pretrained("private/model", token=access_token)
 
 def get_prompt(message, chat_history, system_prompt):