RohitGandikota commited on
Commit
7a90cda
·
verified ·
1 Parent(s): ca2e35f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  import torch
8
- from diffusers import DiffusionPipeline, UNet2DConditionModel, LCMScheduler, AutoencoderTiny
9
  from huggingface_hub import hf_hub_download
10
  from safetensors.torch import load_file
11
  import sys
@@ -634,7 +634,7 @@ if __name__ == "__main__":
634
  # value=4, # Replace with defaults that work for your model
635
  # )
636
 
637
- # # gr.minoExamples(examples=examples, inputs=[prompt])
638
  # gr.on(
639
  # triggers=[run_button.click, prompt.submit],
640
  # fn=infer,
 
5
  import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
7
  import torch
8
+ from diffusers import DiffusionPipeline, UNet2DConditionModel, LCMScheduler, AutoencoderTiny, FluxPipeline
9
  from huggingface_hub import hf_hub_download
10
  from safetensors.torch import load_file
11
  import sys
 
634
  # value=4, # Replace with defaults that work for your model
635
  # )
636
 
637
+ # # gr.Examples(examples=examples, inputs=[prompt])
638
  # gr.on(
639
  # triggers=[run_button.click, prompt.submit],
640
  # fn=infer,