File size: 856 Bytes
b935e3d f404d24 36bfcb2 b935e3d 5dc1a58 36bfcb2 5dc1a58 52e14d6 5dc1a58 f404d24 36bfcb2 ed0ae5f 5dc1a58 ed0ae5f 5dc1a58 52e14d6 ed0ae5f f404d24 |
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 |
import gradio as gr
from transformers import pipeline
from spaces import GPU # Import the GPU decorator for ZeroGPU
# Use @GPU to allocate a GPU when running the video generation function
@GPU
def generate_video(prompt):
# Load the Kandinsky 1.1 video model
video_model = pipeline("video-generation", model="ai-forever/KandinskyVideo_1_1", device=0)
# Generate video based on the prompt
video = video_model(prompt)
return video # Ensure this returns the video file or frames
# Create the Gradio interface
interface = gr.Interface(
fn=generate_video,
inputs=gr.Textbox(label="Enter your prompt here"),
outputs="video",
title="AI Video Generator",
description="This app generates a short video based on your input prompt using Kandinsky 1.1.",
theme="dark"
)
if __name__ == "__main__":
interface.launch()
|