tricktreat commited on
Commit
c080dc3
1 Parent(s): a948da3
Files changed (2) hide show
  1. README.md +1 -1
  2. awesome_chat.py +2 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 😻
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
7
- sdk_version: 3.24.1
8
  app_file: app.py
9
  pinned: false
10
  ---
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
7
+ sdk_version: 3.28.3
8
  app_file: app.py
9
  pinned: false
10
  ---
awesome_chat.py CHANGED
@@ -795,7 +795,8 @@ def run_task(input, command, results, openaikey = None, huggingfacetoken = None)
795
  inference_result = {"error": f"no available models on {command['task']} task."}
796
  results[id] = collect_result(command, "", inference_result)
797
  return False
798
-
 
799
  if len(all_avaliable_model_ids) == 1:
800
  best_model_id = all_avaliable_model_ids[0]
801
  hosted_on = "local" if best_model_id in all_avaliable_models["local"] else "huggingface"
795
  inference_result = {"error": f"no available models on {command['task']} task."}
796
  results[id] = collect_result(command, "", inference_result)
797
  return False
798
+
799
+ all_avaliable_model_ids = all_avaliable_model_ids[:1]
800
  if len(all_avaliable_model_ids) == 1:
801
  best_model_id = all_avaliable_model_ids[0]
802
  hosted_on = "local" if best_model_id in all_avaliable_models["local"] else "huggingface"