Create app.py

app.py

@@ -0,0 +1,50 @@
+# Устанавливаем необходимые библиотеки
+!pip install gradio diffusers torch transformers accelerate scipy numpy librosa
+
+# Импортируем библиотеки
+import gradio as gr
+import torch
+from diffusers import StableVideoDiffusionPipeline
+from transformers import MusicgenForConditionalGeneration, MusicgenProcessor
+
+# Загрузка модели для генерации видео (Stable Video Diffusion)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+video_pipe = StableVideoDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-video-diffusion-img2vid",
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+).to(device)
+
+# Загрузка модели для генерации музыки (MusicGen)
+music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
+music_processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
+
+# Функция генерации видео
+def generate_video(prompt):
+    video_frames = video_pipe(prompt, num_inference_steps=25).frames
+    return video_frames
+
+# Функция генерации музыки
+def generate_music(prompt):
+    inputs = music_processor(prompt, return_tensors="pt")
+    audio_output = music_model.generate(**inputs)
+    return audio_output
+
+# Интерфейс Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# OnlyReels.AI 🎥🎵")
+    
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Введите описание видео и музыки", placeholder="Пример: Киберпанк-город ночью, синтезаторная музыка")
+    
+    with gr.Row():
+        btn_video = gr.Button("🎬 Сгенерировать видео")
+        btn_music = gr.Button("🎵 Сгенерировать музыку")
+    
+    video_output = gr.Video(label="Сгенерированное видео")
+    music_output = gr.Audio(label="Сгенерированная музыка")
+    
+    btn_video.click(fn=generate_video, inputs=prompt_input, outputs=video_output)
+    btn_music.click(fn=generate_music, inputs=prompt_input, outputs=music_output)
+
+# Запуск интерфейса
+demo.launch(share=True)