Joeythemonster's picture
Duplicate from ArtGAN/Video-Diffusion-WebUI
4045f37
import gradio as gr
from video_diffusion.damo.damo_text2_video import DamoText2VideoGenerator
from video_diffusion.inpaint_zoom.zoom_in_app import StableDiffusionZoomIn
from video_diffusion.inpaint_zoom.zoom_out_app import StableDiffusionZoomOut
from video_diffusion.stable_diffusion_video.stable_video_text2video import StableDiffusionText2VideoGenerator
from video_diffusion.tuneavideo.tuneavideo_text2video import TunaVideoText2VideoGenerator
from video_diffusion.zero_shot.zero_shot_text2video import ZeroShotText2VideoGenerator
def diffusion_app():
app = gr.Blocks()
with app:
gr.HTML(
"""
<h1 style='text-align: center'>
Video Diffusion WebUI
</h1>
"""
)
gr.HTML(
"""
<h3 style='text-align: center'>
Follow me for more!
<a href='https://twitter.com/kadirnar_ai' target='_blank'>Twitter</a> | <a href='https://github.com/kadirnar' target='_blank'>Github</a> | <a href='https://www.linkedin.com/in/kadir-nar/' target='_blank'>Linkedin</a>
</h3>
"""
)
with gr.Row():
with gr.Column():
with gr.Tab("Stable Diffusion Video"):
StableDiffusionText2VideoGenerator.app()
with gr.Tab("Tune-a-Video"):
TunaVideoText2VideoGenerator.app()
with gr.Tab("Stable Infinite Zoom"):
with gr.Tab("Zoom In"):
StableDiffusionZoomIn.app()
with gr.Tab("Zoom Out"):
StableDiffusionZoomOut.app()
with gr.Tab("Damo Text2Video"):
DamoText2VideoGenerator.app()
with gr.Tab("Zero Shot Text2Video"):
ZeroShotText2VideoGenerator.app()
app.queue(concurrency_count=1)
app.launch(debug=True, enable_queue=True)
if __name__ == "__main__":
diffusion_app()