TextToVideo / app.py
Rooni's picture
Create app.py
f5076e4
import gradio as gr
from transformers import pipeline
# Загрузка модели Text2video
text2video = pipeline("text-to-video-generation", model="hotshotco/Hotshot-XL")
def generate_video(text):
# Генерация видео из текста с использованием модели
video = text2video(text)
# Возвращаем результат в видео-файла
return video['video_file']
# Создание интерфейса с использованием gr.Blocks
with gr.Blocks() as demo:
with gr.Row():
with gr.Column(scale=2):
text_input = gr.Textbox(label="Введите текст", placeholder="Введите описание видео здесь...")
with gr.Column():
video_output = gr.Video(label="Сгенерированное видео")
generate_button = gr.Button("Сгенерировать видео")
generate_button.click(fn=generate_video, inputs=text_input, outputs=video_output)
# Запуск интерфейса
demo.launch()