Update app.py

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 import numpy as np
 
 pipeline = DiffusionPipeline.from_pretrained("1aurent/ddpm-mnist")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = pipeline.to(device=device)
 
 def predict(steps, seed):
     generator = torch.manual_seed(seed)