Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -103,10 +103,7 @@ def refresh_models():
|
|
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 |
|
@@ -134,8 +131,9 @@ def refresh_models():
|
|
134 |
) as executor:
|
135 |
future_to_model = {
|
136 |
executor.submit(
|
137 |
-
|
138 |
-
FREE_MODEL_TEST_KEY,
|
|
|
139 |
): model for model in embedding_models
|
140 |
}
|
141 |
for future in concurrent.futures.as_completed(future_to_model):
|
|
|
103 |
global embedding_models, free_embedding_models
|
104 |
|
105 |
text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
|
106 |
+
embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding") + get_all_models(FREE_MODEL_TEST_KEY, "reranker")
|
|
|
|
|
|
|
107 |
free_text_models = []
|
108 |
free_embedding_models = []
|
109 |
|
|
|
131 |
) as executor:
|
132 |
future_to_model = {
|
133 |
executor.submit(
|
134 |
+
test_model_availability,
|
135 |
+
FREE_MODEL_TEST_KEY,
|
136 |
+
model
|
137 |
): model for model in embedding_models
|
138 |
}
|
139 |
for future in concurrent.futures.as_completed(future_to_model):
|