Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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]
|