Update app.py

app.py

@@ -18,12 +18,13 @@ bases = {
 step_loaded = None
 base_loaded = "ToonYou"
 motion_loaded = None
-
+device = "cuda"
 # Ensure model and scheduler are initialized in GPU-enabled function
 if not torch.cuda.is_available():
-    raise NotImplementedError("No GPU detected!")
+    device = "cpu"
+    #raise NotImplementedError("No GPU detected!")
+
 
-device = "cpu"
 dtype = None
 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")