Update app.py

app.py

@@ -0,0 +1,51 @@
+import gradio as gr
+import torch
+from diffusers import StableDiffusionPipeline, AnimateDiffPipeline
+from PIL import Image
+
+# Load the 3D Cartoon Image Generator (DreamShaper)
+image_pipe = StableDiffusionPipeline.from_pretrained(
+    "Lykon/DreamShaper", torch_dtype=torch.float16
+).to("cuda")
+
+# Load the Animation Model (AnimateDiff)
+video_pipe = AnimateDiffPipeline.from_pretrained(
+    "guoyww/animatediff", torch_dtype=torch.float16
+).to("cuda")
+
+# Function to generate a 3D cartoon image
+def generate_cartoon_image(prompt):
+    image = image_pipe(prompt).images[0]
+    image_path = "generated_cartoon.png"
+    image.save(image_path)
+    return image_path
+
+# Function to generate an animated video from an uploaded image
+def generate_cartoon_video(image):
+    image = Image.open(image).convert("RGB")
+    video_output = video_pipe(image)["video"]
+    video_path = "generated_cartoon_animation.mp4"
+    video_output.save(video_path)
+    return video_path
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎨 AI 3D Cartoon Image & Animation Generator")
+    
+    # Text-to-Image Section
+    with gr.Tab("Generate Cartoon Image"):
+        gr.Markdown("Enter a text prompt to generate a 3D cartoon-style image")
+        text_input = gr.Textbox(label="Enter your prompt")
+        img_output = gr.Image(label="Generated Cartoon Image")
+        img_button = gr.Button("Generate Image")
+        img_button.click(generate_cartoon_image, inputs=text_input, outputs=img_output)
+
+    # Image-to-Video Section
+    with gr.Tab("Generate Cartoon Animation"):
+        gr.Markdown("Upload a 3D cartoon-style image to create an animation")
+        img_input = gr.Image(type="filepath", label="Upload Cartoon Image")
+        vid_output = gr.Video(label="Generated Animation")
+        vid_button = gr.Button("Generate Animation")
+        vid_button.click(generate_cartoon_video, inputs=img_input, outputs=vid_output)
+
+demo.launch()