Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
28 |
)
|
29 |
|
30 |
pipe.load_lora_weights("./lora-trained-thumbs-up-pw-0.5-steps-800-uttam-pw-0.5-steps-1600")
|
31 |
-
|
32 |
|
33 |
prompt = st.text_area("Enter the prompt!")
|
34 |
image = pipe(prompt=prompt, num_inference_steps=25, num_images_per_prompt=1)[:]
|
|
|
28 |
)
|
29 |
|
30 |
pipe.load_lora_weights("./lora-trained-thumbs-up-pw-0.5-steps-800-uttam-pw-0.5-steps-1600")
|
31 |
+
_ = pipe.to("cuda")
|
32 |
|
33 |
prompt = st.text_area("Enter the prompt!")
|
34 |
image = pipe(prompt=prompt, num_inference_steps=25, num_images_per_prompt=1)[:]
|