aai / old2 /app.py
barreloflube's picture
Refactor flux_helpers.py to enable or disable Vae
37112ef
import gradio as gr
from config import css
from tabs.image_tab import image_tab
from tabs.audio_tab import audio_tab
from tabs.video_tab import video_tab
from tabs.text_tab import text_tab
def main():
with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
# Header
with gr.Column(elem_classes="center-content"):
gr.Markdown("""
# πŸš€ AAI: All AI
Unleash your creativity with our multi-modal AI platform.
[![Sync code to HF Space](https://github.com/mantrakp04/aai/actions/workflows/hf-space.yml/badge.svg)](https://github.com/mantrakp04/aai/actions/workflows/hf-space.yml)
""")
# Tabs
with gr.Tabs():
with gr.Tab(label="πŸ–ΌοΈ Image"):
image_tab()
with gr.Tab(label="🎡 Audio"):
audio_tab()
with gr.Tab(label="πŸŽ₯ Video"):
video_tab()
with gr.Tab(label="πŸ“ Text"):
text_tab()
demo.launch(
share=False,
debug=True,
)
if __name__ == "__main__":
main()