video model

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 from spaces import GPU  # Import the GPU decorator for ZeroGPU
-from PIL import Image
 
 # Use @GPU to allocate a GPU when running the video generation function
 @GPU
 def generate_video(prompt):
-    # Load the Kandinsky video model
+    # 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)
@@ -18,7 +17,7 @@ interface = gr.Interface(
     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 KandinskyVideo_1_1.",
+    description="This app generates a short video based on your input prompt using Kandinsky 1.1.",
     theme="dark"
 )