Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
+
)
|