Update app.py

app.py

@@ -109,10 +109,6 @@ def generate(
     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)
@@ -210,8 +206,8 @@ def launch_app(gen_fn: Callable):
 # ------------------------------------------------------------------------------------- #
 
 diffuser, clip_model, dataset, device = init("config")
-diffuser.to(device)
-clip_model.to(device)
+diffuser.to("cuda")
+clip_model.to("cuda")
 
 generate_sample = partial(
     generate,