File size: 3,572 Bytes
9aa6aea a7c6ff5 ce50c4e 9aa6aea d4c4fe6 9aa6aea 123074d 7e88e86 9aa6aea a7c6ff5 9aa6aea bad108e 1b6ac29 0a9d7de a20173d 1b6ac29 a7c6ff5 1b6ac29 0a9d7de 1b6ac29 0a9d7de 1b6ac29 bad108e 9aa6aea 1b6ac29 9aa6aea 0a9d7de 9aa6aea 6e90f61 9aa6aea cd9f1f6 9aa6aea cd9f1f6 9aa6aea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
# -*- coding: utf-8 -*-
# ===================================================
#
# Author : Fan Zhang
# Email : zhangfan@baai.ac.cn
# Institute : Beijing Academy of Artificial Intelligence (BAAI)
# Create On : 2023-12-11 15:34
# Last Modified : 2024-01-04 10:15
# File Name : app.py
# Description :
#
# ===================================================
import os
os.system("pip uninstall -y gradio")
os.system("pip install gradio==3.40.1")
import argparse
import gradio as gr
from demo.generation_frontend import build_generation
from demo.chat_frontend import build_chat
parser = argparse.ArgumentParser()
parser.add_argument("--title", type=str, default='Emu')
parser.add_argument("--host", type=str, default="0.0.0.0")
parser.add_argument("--port", type=int, default=9002)
parser.add_argument("--share", action="store_true")
parser.add_argument("--controller-url", type=str, default="http://218.91.113.230:9003")
parser.add_argument("--concurrency-count", type=int, default=8)
parser.add_argument("--disable-chat", action="store_true")
parser.add_argument("--disable-generate", action="store_true")
args = parser.parse_args()
# <h2 align='center'>[<a href='https://emu.ssi.plus' target='_blank' rel='noopener'>FAST demo</a>]</h2>
if __name__ == "__main__":
title = "Emu2: Generative Multimodal Models are In-Context Learners<br> \
<h2 align='center'> \
[<a href='https://baaivision.github.io/emu2' target='_blank' rel='noopener'>project page</a>] \
[<a href='https://github.com/baaivision/Emu' target='_blank' rel='noopener'>code</a>] \
[<a href='https://arxiv.org/abs/2312.13286' target='_blank' rel='noopener'>paper</a>] \
</h2> \
<div align='center'> \
<font size=4>Experience Emu2 with 👉</font> \
<font size=6>[<b><a href='https://emu.ssi.plus' target='_blank' rel='noopener'>FAST demo</a>]</b></font> \
<font size=4>👈 <b>WAY MORE FASTER!!!</b></font> \
</div> \
<h3 align='center'> \
🤗HF models: \
<a href='https://huggingface.co/BAAI/Emu2' target='_blank' rel='noopener'>Emu2</a> | \
<a href='https://huggingface.co/BAAI/Emu2-Chat' target='_blank' rel='noopener'>Emu2-Chat</a> | \
<a href='https://huggingface.co/BAAI/Emu2-Gen' target='_blank' rel='noopener'>Emu2-Gen</a> \
</h3> \
<h4 align='center'> \
[<a href='https://jwolpxeehx.feishu.cn/docx/KskPdU99FomufKx4G9hcQMeQnHv' target='_blank' rel='noopener'>使用说明</a>] \
[<a href='https://jwolpxeehx.feishu.cn/docx/RYHNd1tvEo8k8Mx9HeMcvvxWnvZ' target='_blank' rel='noopener'>User Guide</a>] \
</h4> \
"
interface_list, tab_names = [], []
if not args.disable_chat:
demo_chat = build_chat(args)
interface_list.append(demo_chat)
tab_names.append("Multimodal Chat")
if not args.disable_generate:
demo_generation = build_generation(args)
interface_list.append(demo_generation)
tab_names.append("Multimodal Generation")
demo_all = gr.TabbedInterface(
interface_list=interface_list,
tab_names=tab_names,
title=title,
theme=gr.themes.Default(primary_hue="blue", secondary_hue="blue"),
)
demo_all.queue(
concurrency_count=args.concurrency_count,
status_update_rate=3,
api_open=False,
).launch(
enable_queue=True,
share=args.share,
)
|