Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+iimport gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 from moviepy.editor import CompositeVideoClip, ImageClip, AudioFileClip, VideoFileClip, concatenate_videoclips
@@ -7,10 +7,11 @@ import torch
 from PIL import Image
 import numpy as np
 from scipy.io.wavfile import write
+import random
 
 # Load and Initialize Models
 script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
-image_generator = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cuda")
+image_generator = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cpu")
 music_generator = pipeline("text-to-music", model="HuggingFace/MusicGen")  # Replace with actual model if available
 
 # Generate Comedy Script
@@ -132,3 +133,4 @@ app.launch()
 
 
 
+