Upload app.py

app.py

@@ -75,7 +75,7 @@ def change_base_model(repo_id: str, cn_on: bool): # , progress=gr.Progress(track
             print(f"Loading model: {repo_id}")
             #clear_cache()
             pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=dtype, vae=taef1).to(device)
-            #pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
+            pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
             last_model = repo_id
             last_cn_on = cn_on
             #progress(1, desc=f"Model loaded: {repo_id}")