amitashukla commited on
Commit
4228096
·
verified ·
1 Parent(s): 9f96a54
Files changed (1) hide show
  1. src/chat.py +1 -1
src/chat.py CHANGED
@@ -71,7 +71,7 @@ class Chatbot:
71
  Initialize the chatbot with a HF model ID
72
  """
73
  model_id = MY_MODEL if MY_MODEL else BASE_MODEL # define MY_MODEL in config.py if you create a new model in the HuggingFace Hub
74
- self.client = InferenceClient(model=model_id, token="HF_TOKEN")
75
  # Initialize user profile
76
  current_dir = os.path.dirname(os.path.abspath(__file__))
77
  data_dir = os.path.join(current_dir, '..', 'data')
 
71
  Initialize the chatbot with a HF model ID
72
  """
73
  model_id = MY_MODEL if MY_MODEL else BASE_MODEL # define MY_MODEL in config.py if you create a new model in the HuggingFace Hub
74
+ self.client = InferenceClient(model=model_id, token=HF_TOKEN)
75
  # Initialize user profile
76
  current_dir = os.path.dirname(os.path.abspath(__file__))
77
  data_dir = os.path.join(current_dir, '..', 'data')