| import gradio as gr |
| import requests |
|
|
| HF_API_TOKEN = "YOUR_HF_API_TOKEN" |
| API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1" |
|
|
| headers = {"Authorization": f"Bearer {HF_API_TOKEN}"} |
|
|
| def query(payload): |
| response = requests.post(API_URL, headers=headers, json=payload) |
| return response.content |
|
|
| def generate_video(prompt): |
| os.makedirs("frames", exist_ok=True) |
| for i in range(10): |
| image_bytes = query({"inputs": prompt}) |
| with open(f"frames/frame_{i:03d}.png", "wb") as f: |
| f.write(image_bytes) |
|
|
| create_video_from_frames("frames", "generated_video.mp4", fps=2) |
| return "generated_video.mp4" |
|
|
| def create_video_from_frames(frame_folder, output_path, fps=2): |
| import cv2 |
| images = sorted([img for img in os.listdir(frame_folder) if img.endswith(".png")]) |
| frame = cv2.imread(os.path.join(frame_folder, images[0])) |
| height, width, _ = frame.shape |
| video = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height)) |
| for img in images: |
| video.write(cv2.imread(os.path.join(frame_folder, img))) |
| video.release() |
|
|
| iface = gr.Interface(fn=generate_video, |
| inputs=gr.Textbox(lines=3), |
| outputs=gr.Video(), |
| title="Text to Video AI") |
|
|
| if __name__ == "__main__": |
| iface.launch() |
|
|
