File size: 534 Bytes
a40b44d debb93e c00fe36 c364545 6dc0c9c c00fe36 73237fe 6dc0c9c 73237fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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() |