linoyts HF Staff commited on
Commit
7dbbdc6
·
verified ·
1 Parent(s): 6b5fd4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,7 +6,8 @@ import torch
6
  import random
7
  from PIL import Image
8
 
9
- from kontext_pipeline import FluxKontextPipeline
 
10
  from diffusers import FluxTransformer2DModel
11
  from diffusers.utils import load_image
12
 
 
6
  import random
7
  from PIL import Image
8
 
9
+ #from kontext_pipeline import FluxKontextPipeline
10
+ from pipeline_flux_kontext import FluxKontextPipeline
11
  from diffusers import FluxTransformer2DModel
12
  from diffusers.utils import load_image
13