Update app.py

app.py

@@ -48,7 +48,7 @@ def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_ste
     inputs = tokenizer(prompt, return_tensors="pt").input_ids
     with torch.no_grad():
       outputs = model.generate(
-        inputs,
+        inputs.to(device),
         do_sample=True,
         temperature=1.0,
         top_p=1.0,