KingNish commited on
Commit
3ce6a82
1 Parent(s): 169bac6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -113,13 +113,13 @@ def king(type ,
113
  if randomize_seed:
114
  seed = random.randint(0, 99999)
115
  generator = torch.Generator().manual_seed(seed)
116
- if style=="3D"
117
  image = pipe_3D( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
118
- elif style=="PixelArt"
119
  image = pipe_pixel( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
120
- elif style=="Logo"
121
  image = pipe_logo( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
122
- elif style=="Origami"
123
  image = pipe_ori( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
124
  else:
125
  image = pipe_best( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
 
113
  if randomize_seed:
114
  seed = random.randint(0, 99999)
115
  generator = torch.Generator().manual_seed(seed)
116
+ if style=="3D":
117
  image = pipe_3D( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
118
+ elif style=="PixelArt":
119
  image = pipe_pixel( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
120
+ elif style=="Logo":
121
  image = pipe_logo( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
122
+ elif style=="Origami":
123
  image = pipe_ori( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]
124
  else:
125
  image = pipe_best( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]