Update app.py

app.py

@@ -21,13 +21,11 @@ step_loaded = None
 base_loaded = "Realistic"
 motion_loaded = None
 
-# Set device to CPU
 device = "cpu"
 dtype = torch.float32
 pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
 
-@spaces.CPU(duration=30, queue=False)
 def generate_image(prompt, base="Realistic", motion="", step=8, duration=5, progress=gr.Progress()):
     global step_loaded
     global base_loaded
@@ -120,4 +118,4 @@ with gr.Blocks() as demo:
         outputs=[video]
     )
 
-demo.queue().launch()
+demo.queue().launch()