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__": | |
main() |