Update app.py
Browse files
app.py
CHANGED
@@ -8,13 +8,11 @@ dtype = torch.float16 if device == "cuda" else torch.float32
|
|
8 |
variant = "fp16" if device == "cuda" else None
|
9 |
|
10 |
pipe = DiffusionPipeline.from_pretrained(
|
11 |
-
"stabilityai/
|
12 |
torch_dtype=dtype,
|
13 |
variant=variant
|
14 |
).to(device)
|
15 |
|
16 |
-
pipe.load_lora_weights("artificialguybr/TshirtDesignRedmond-V2")
|
17 |
-
|
18 |
def infer(color_prompt, phone_type_prompt, design_prompt):
|
19 |
prompt = (
|
20 |
f"A single vertical {color_prompt} colored {phone_type_prompt} back cover featuring a bold {design_prompt} design on the front, hanging on the plain wall. The soft light and shadows, creating a striking contrast against the minimal background, evoking modern sophistication."
|
|
|
8 |
variant = "fp16" if device == "cuda" else None
|
9 |
|
10 |
pipe = DiffusionPipeline.from_pretrained(
|
11 |
+
"stabilityai/sdxl-turbo",
|
12 |
torch_dtype=dtype,
|
13 |
variant=variant
|
14 |
).to(device)
|
15 |
|
|
|
|
|
16 |
def infer(color_prompt, phone_type_prompt, design_prompt):
|
17 |
prompt = (
|
18 |
f"A single vertical {color_prompt} colored {phone_type_prompt} back cover featuring a bold {design_prompt} design on the front, hanging on the plain wall. The soft light and shadows, creating a striking contrast against the minimal background, evoking modern sophistication."
|