update

app.py

@@ -120,7 +120,7 @@ def change_model_fn(model_name: str) -> None:
             torch_dtype=torch.bfloat16,
             feature_extractor=None,
             safety_checker=None
-        )
+        ).to(device)
         pipeline = ConceptrolIPAdapterPlus(pipe, "", adapter_name, device, num_tokens=16)
         globals()["pipeline"] = pipeline
     else:
@@ -202,6 +202,7 @@ def generate(
 ) -> np.ndarray:
     global pipeline
     change_model_fn(model_name)
+    print(image)
     if isinstance(pipeline, FluxConceptrolPipeline):
         images = pipeline(
             prompt=prompt,