import spaces # import first
from loguru import logger
logger.info("app start")
import gradio as gr

from src.model.model_llava import inference_by_prompt_and_images, inference_by_prompt_and_images_fire # keep it here
from src.serve.gradio_block_arena_vision_named import build_side_by_side_vision_ui_named

def main():
    with gr.Blocks() as demo:
        states = build_side_by_side_vision_ui_named(
            models=["llava-fire", "llava-original"]
        )
        
        demo.launch()

if __name__ == "__main__":
    logger.info("in main")
    main()