Hugging Face's logo

Spaces:
WAT-ai-AA
/
stable-diffused-adversarial-attacks
Runtime error

App Files Community
1
harshp3030 commited on
Commit
948c263
·
1 Parent(s): b632ef5

model_id definition

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
3
  import torch
4
 
 
5
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
6
 
7
  def txt2img(prompt):
 
2
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
3
  import torch
4
 
5
+ model_id = "stabilityai/stable-diffusion-2"
6
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
7
 
8
  def txt2img(prompt):