gradio.py

import gradio as gr
from hydra import compose, initialize
from omegaconf import OmegaConf

from .src.demo import App

def main():
    with initialize(version_base=None, config_path="conf", job_name="test_app"):
        cfg = compose(config_name="config", overrides=["db=mysql", "db.user=me"])
        
        app = App(cfg)

        webapp = gr.ChatInterface(fn=app.ask_chat, examples=["hello", "hola", "merhaba"], title="LLM4SciLit")
        webapp.launch(share=True)

if __name__ == "__main__":
    main()