dvoils commited on
Commit
1161619
1 Parent(s): 15a1b0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from diffusers import DDPMPipeline
3
  import PIL
4
 
5
  # Use the cache decorator to only re-run when the inputs change
6
- @st.cache(suppress_st_warning=True, allow_output_mutation=True)
7
  def generate_image():
8
  pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
9
  image = pipeline().images[0]
 
3
  import PIL
4
 
5
  # Use the cache decorator to only re-run when the inputs change
6
+ # @st.cache(suppress_st_warning=True, allow_output_mutation=True)
7
  def generate_image():
8
  pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
9
  image = pipeline().images[0]