Hugging Face's logo

Spaces:
Bagda
/
Imagen
Runtime error

App Files Community
seawolf2357 commited on
Commit
acd2b67
·
verified ·
1 Parent(s): a9d2d31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from compel import Compel, ReturnedEmbeddingsType
6
  from diffusers import DiffusionPipeline
7
 
8
  # Load model
9
- model_name = os.environ.get('MODEL_NAME', 'UnfilteredAI/NSFW-gen-v2')
10
  pipe = DiffusionPipeline.from_pretrained(
11
  model_name,
12
  torch_dtype=torch.float16
 
6
  from diffusers import DiffusionPipeline
7
 
8
  # Load model
9
+ model_name = os.environ.get('MODEL_NAME', 'UnfilteredAI/NSFW-gen-v2.1')
10
  pipe = DiffusionPipeline.from_pretrained(
11
  model_name,
12
  torch_dtype=torch.float16