Update app.py

app.py

@@ -17,7 +17,7 @@ def predict(steps, seed, value, guidance):
     for i in range(1,steps):
         yield pipeline(
             generator=generator,
-            condition=torch.tensor([value]),
+            condition=torch.tensor([value, device=device]),
             guidance=guidance,
             num_inference_steps=steps
         ).images[0]