Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 import numpy as np
 import random
-
-# import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
-import torch
+
+pipe = DiffusionPipeline.from_pretrained("THUDM/CogVideoX-5b")
+
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+image = pipe(prompt).images[0]
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use