Update app.py

app.py

@@ -1,24 +1,26 @@
+
+import torch
+from diffusers import StableDiffusionPipeline
 import gradio as gr
 
-# Dummy function (replace this with your actual Zeroscope video generation logic)
-def generate_video(prompt):
-    # You can replace this string with actual file path or video URL
-    return f"🎬 Video generated from prompt: '{prompt}'"
+# Load Stable Diffusion on CPU
+pipe = StableDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-2-1-base",
+    torch_dtype=torch.float32
+).to("cpu")
+pipe.enable_attention_slicing()
+
+# Function for generation
+def generate(prompt):
+    image = pipe(prompt, num_inference_steps=15, guidance_scale=7.5).images[0]
+    return image
 
-# Gradio Blocks UI with API exposure
-with gr.Blocks() as demo:
-    gr.Markdown("## 🎥 Zeroscope Video Generator")
-    
-    with gr.Row():
-        prompt = gr.Textbox(label="Enter Prompt", placeholder="Type your video idea here...")
-    
-    with gr.Row():
-        output = gr.Textbox(label="Result")
-    
-    submit = gr.Button("Generate Video")
-    
-    # Expose this to API by naming the endpoint
-    submit.click(fn=generate_video, inputs=prompt, outputs=output, api_name="generate")
+# Gradio Interface
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(label="Prompt"),
+    outputs=gr.Image(type="pil"),
+)
 
-# Launch with API docs enabled
-demo.queue().launch(show_api=True)
+# Launch Space
+demo.launch()