Spaces:
Runtime error
Runtime error
script with no condition
Browse files
src/inference_no_condition.py
CHANGED
@@ -17,7 +17,7 @@ pipe.to("cuda")
|
|
17 |
with open('../examples/prompts.json', 'r') as f:
|
18 |
prompts_list = list(json.load(f).values())
|
19 |
|
20 |
-
image_index =
|
21 |
prompt, negative_prompt = prompts_list[image_index]
|
22 |
|
23 |
images = pipe(
|
@@ -29,6 +29,6 @@ images = pipe(
|
|
29 |
num_inference_steps=20,
|
30 |
guidance_scale=9.0,
|
31 |
).images[0]
|
32 |
-
|
33 |
plt.imshow(images)
|
34 |
plt.show()
|
|
|
17 |
with open('../examples/prompts.json', 'r') as f:
|
18 |
prompts_list = list(json.load(f).values())
|
19 |
|
20 |
+
image_index = 4
|
21 |
prompt, negative_prompt = prompts_list[image_index]
|
22 |
|
23 |
images = pipe(
|
|
|
29 |
num_inference_steps=20,
|
30 |
guidance_scale=9.0,
|
31 |
).images[0]
|
32 |
+
images.save(fr'C:\Users\dragynir\Pictures\FASHION_IMAGES\no_condition\image_{image_index+1}.png')
|
33 |
plt.imshow(images)
|
34 |
plt.show()
|