FIRE / app.py
zhangbofei
fix: fire
c364545
raw
history blame
534 Bytes
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()