Update app.py

app.py

@@ -108,6 +108,11 @@ def generate(
     # Inference
     sample_id = SAMPLE_IDS[LABEL_TO_IDS[sample_label]]
     seq_feat = diffuser.net.model.clip_sequential
+
+    print(device)
+    print(clip_model.device)
+    print(diffuser.device)
+    
     batch = get_batch(prompt, sample_id, clip_model, dataset, seq_feat, device)
     with torch.no_grad():
         out = diffuser.predict_step(batch, 0)