Spaces:
Runtime error
Runtime error
| # -*- coding: utf-8 -*- | |
| # =================================================== | |
| # | |
| # Author : Fan Zhang | |
| # Email : [email protected] | |
| # Institute : Beijing Academy of Artificial Intelligence (BAAI) | |
| # Create On : 2023-12-11 15:34 | |
| # Last Modified : 2023-12-21 02:25 | |
| # 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=2) | |
| parser.add_argument("--disable-chat", action="store_true") | |
| parser.add_argument("--disable-generate", action="store_true") | |
| args = parser.parse_args() | |
| 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> \ | |
| <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_generate: | |
| demo_generation = build_generation(args) | |
| interface_list.append(demo_generation) | |
| tab_names.append("Multimodal Generation") | |
| if not args.disable_chat: | |
| demo_chat = build_chat(args) | |
| interface_list.append(demo_chat) | |
| tab_names.append("Multimodal Chat") | |
| 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, | |
| ) | |