Spaces:
Running
Running
Jae-Won Chung
commited on
Commit
·
aaadf66
1
Parent(s):
9bd3aab
Allow disabling Colosseum for local testing
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ pio.templates.default = "plotly_white"
|
|
| 23 |
from spitfight.colosseum.client import ControllerClient
|
| 24 |
|
| 25 |
COLOSSEUM_UP = True
|
| 26 |
-
|
| 27 |
|
| 28 |
|
| 29 |
class TableManager:
|
|
@@ -371,7 +371,11 @@ Every benchmark is limited in some sense -- Before you interpret the results, pl
|
|
| 371 |
"""
|
| 372 |
|
| 373 |
# The app will not start without a controller address set.
|
| 374 |
-
controller_addr = os.environ
|
|
|
|
|
|
|
|
|
|
|
|
|
| 375 |
global_controller_client = ControllerClient(controller_addr=controller_addr, timeout=15)
|
| 376 |
|
| 377 |
# Load the list of models. To reload, the app should be restarted.
|
|
@@ -541,7 +545,7 @@ with gr.Blocks(css=custom_css) as block:
|
|
| 541 |
if COLOSSEUM_UP:
|
| 542 |
gr.Markdown(open("docs/colosseum_top.md").read())
|
| 543 |
else:
|
| 544 |
-
gr.HTML(
|
| 545 |
gr.HTML("<h3 style='text-align: center'>The energy leaderboard is still available.</h3><br/>")
|
| 546 |
|
| 547 |
with gr.Row():
|
|
|
|
| 23 |
from spitfight.colosseum.client import ControllerClient
|
| 24 |
|
| 25 |
COLOSSEUM_UP = True
|
| 26 |
+
COLOSSEUM_DOWN_MESSAGE = f"<br/><h2 style='text-align: center'>The Colosseum is currently down. We'll be back on <u><b>September 4th</b></u>.</h2>"
|
| 27 |
|
| 28 |
|
| 29 |
class TableManager:
|
|
|
|
| 371 |
"""
|
| 372 |
|
| 373 |
# The app will not start without a controller address set.
|
| 374 |
+
controller_addr = os.environ.get("COLOSSEUM_CONTROLLER_ADDR")
|
| 375 |
+
if controller_addr is None:
|
| 376 |
+
COLOSSEUM_UP = False
|
| 377 |
+
COLOSSEUM_DOWN_MESSAGE = "<br/><h2 style='text-align: center'>Disabled Colosseum for local testing.</h2>"
|
| 378 |
+
controller_addr = "localhost"
|
| 379 |
global_controller_client = ControllerClient(controller_addr=controller_addr, timeout=15)
|
| 380 |
|
| 381 |
# Load the list of models. To reload, the app should be restarted.
|
|
|
|
| 545 |
if COLOSSEUM_UP:
|
| 546 |
gr.Markdown(open("docs/colosseum_top.md").read())
|
| 547 |
else:
|
| 548 |
+
gr.HTML(COLOSSEUM_DOWN_MESSAGE)
|
| 549 |
gr.HTML("<h3 style='text-align: center'>The energy leaderboard is still available.</h3><br/>")
|
| 550 |
|
| 551 |
with gr.Row():
|