soiz1 commited on
Commit
518e038
·
verified ·
1 Parent(s): 8717ff4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import rembg
4
  import random
5
  import gradio as gr
6
  import numpy
7
-
 
8
  from PIL import Image
9
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
10
 
 
4
  import random
5
  import gradio as gr
6
  import numpy
7
+ import os
8
+ os.system("sed -i 's/jax\.tree_util\.register_dataclass/jax\.tree_util\.register_pytree_node/g' /usr/local/lib/python3.11/dist-packages/flax/struct.py")
9
  from PIL import Image
10
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
11