yangtb24 commited on
Commit
3c0d2e2
1 Parent(s): 7c381b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -103,7 +103,10 @@ def refresh_models():
103
  global embedding_models, free_embedding_models
104
 
105
  text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
106
- embedding_list = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
 
 
 
107
  free_text_models = []
108
  free_embedding_models = []
109
 
@@ -940,4 +943,4 @@ if __name__ == '__main__':
940
  debug=False,
941
  host='0.0.0.0',
942
  port=int(os.environ.get('PORT', 7860))
943
- )
 
103
  global embedding_models, free_embedding_models
104
 
105
  text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
106
+ embedding_models = {
107
+ sub_type: get_all_models(FREE_MODEL_TEST_KEY, sub_type)
108
+ for sub_type in ["embedding", "reranker"]
109
+ }
110
  free_text_models = []
111
  free_embedding_models = []
112
 
 
943
  debug=False,
944
  host='0.0.0.0',
945
  port=int(os.environ.get('PORT', 7860))
946
+ )