fix
Browse files- 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=
|
| 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')
|