update gradio_web_server.py
Browse files- gradio_web_server.py +3 -0
gradio_web_server.py
CHANGED
|
@@ -62,9 +62,12 @@ def sort_models(models):
|
|
| 62 |
|
| 63 |
|
| 64 |
def get_model_list():
|
|
|
|
| 65 |
ret = requests.post(args.controller_url + "/refresh_all_workers")
|
|
|
|
| 66 |
assert ret.status_code == 200
|
| 67 |
ret = requests.post(args.controller_url + "/list_models")
|
|
|
|
| 68 |
models = ret.json()["models"]
|
| 69 |
models = sort_models(models)
|
| 70 |
|
|
|
|
| 62 |
|
| 63 |
|
| 64 |
def get_model_list():
|
| 65 |
+
logger.info(f"Call `get_model_list`")
|
| 66 |
ret = requests.post(args.controller_url + "/refresh_all_workers")
|
| 67 |
+
logger.info(f"status_code from `get_model_list`: {ret.status_code}")
|
| 68 |
assert ret.status_code == 200
|
| 69 |
ret = requests.post(args.controller_url + "/list_models")
|
| 70 |
+
logger.info(f"status_code from `list_models`: {ret.status_code}")
|
| 71 |
models = ret.json()["models"]
|
| 72 |
models = sort_models(models)
|
| 73 |
|