Update awesome_chat.py

awesome_chat.py

@@ -19,12 +19,7 @@ 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,  # Make sure this matches the function name in get_token_ids.py
-    count_tokens,
-    get_max_context_length
-)
+from get_token_ids import get_token_ids_for_task_parsing, get_token_ids_for_choose_model, 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