htr_demo / app.py
Gabriel's picture
fixed bug with utils for loading db
3810c45
raw
history blame
1.37 kB
import uuid
import gradio as gr
from helper.gradio_config import css, theme
from helper.text.text_app import TextApp
from helper.utils import TrafficDataHandler
from tabs.about_tab import about_tab
from tabs.help_tab import help_tab
from tabs.htr_tool import htr_tool_tab
from tabs.stepwise_htr_tool import stepwise_htr_tool_tab
session_uuid = str(uuid.uuid1())
VERSION = "Demo version 0.0.2"
with gr.Blocks(title="Riksarkivet", theme=theme, css=css) as demo:
with gr.Row():
with gr.Column(scale=1):
text_ip_output = gr.Markdown(VERSION)
with gr.Column(scale=1):
gr.Markdown(TextApp.title_markdown)
with gr.Column(scale=1):
gr.Markdown(TextApp.title_markdown_img)
with gr.Tabs():
with gr.Tab("Fast track"):
htr_tool_tab.render()
with gr.Tab("Stepwise"):
stepwise_htr_tool_tab.render()
with gr.Tab("Help"):
help_tab.render()
with gr.Tab("About"):
about_tab.render()
# SECRET_KEY = os.environ.get("AM_I_IN_A_DOCKER_CONTAINER", False)
# if SECRET_KEY:
demo.load(fn=TrafficDataHandler.onload_store_metric_data, inputs=None, outputs=None)
demo.queue(concurrency_count=2, max_size=2)
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False, show_error=True)