Update app.py

app.py

@@ -101,9 +101,15 @@ def refresh_models():
     """
     global text_models, free_text_models
     global embedding_models, free_embedding_models
-    
-    text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
-    embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding") + get_all_models(FREE_MODEL_TEST_KEY, "reranker")
+
+    text_models = get_all_models(
+        FREE_MODEL_TEST_KEY,
+        "chat"
+    )
+    embedding_models = get_all_models(
+        FREE_MODEL_TEST_KEY,
+        "embedding"
+    ) + get_all_models(FREE_MODEL_TEST_KEY, "reranker")
     free_text_models = []
     free_embedding_models = []
 
@@ -117,7 +123,9 @@ def refresh_models():
                 model
             ): model for model in text_models
         }
-        for future in concurrent.futures.as_completed(future_to_model):
+        for future in concurrent.futures.as_completed(
+            future_to_model
+        ):
             model = future_to_model[future]
             try:
                 is_free = future.result()
@@ -131,12 +139,14 @@ def refresh_models():
     ) as executor:
         future_to_model = {
             executor.submit(
-                test_model_availability,
+                test_embedding_model_availability,
                 FREE_MODEL_TEST_KEY,
                 model
             ): model for model in embedding_models
         }
-        for future in concurrent.futures.as_completed(future_to_model):
+        for future in concurrent.futures.as_completed(
+            future_to_model
+        ):
             model = future_to_model[future]
             try:
                 is_free = future.result()