Update app.py

app.py

@@ -22,12 +22,15 @@ step_loaded = None
 base_loaded = "Realistic"
 motion_loaded = None
 
-# Ensure model and scheduler are initialized in GPU-enabled function
-if not torch.cuda.is_available():
-    raise NotImplementedError("No GPU detected!")
+# Device selection
+if torch.cuda.is_available():
+    device = "cuda"
+    dtype = torch.float16
+else:
+    print("⚠️ No GPU detected! Falling back to CPU. Inference will be slow.")
+    device = "cpu"
+    dtype = torch.float32
 
-device = "cuda"
-dtype = torch.float16
 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")