Update awesome_chat.py

awesome_chat.py

@@ -19,7 +19,12 @@ from diffusers.utils import load_image
 from pydub import AudioSegment
 import threading
 from queue import Queue
-from get_token_ids import get_token_ids_for_task_parsing, get_token_ids_for_choose_model, count_tokens, get_max_context_length
+from get_token_ids import (
+    get_token_ids_for_task_parsing,
+    get_token_ids_for_choose_model,  # Make sure this matches the function name in get_token_ids.py
+    count_tokens,
+    get_max_context_length
+)
 from huggingface_hub.inference_api import InferenceApi
 from huggingface_hub.inference_api import ALL_TASKS
 from models_server import models, status