gaur3009 commited on
Commit
883f499
·
verified ·
1 Parent(s): 92d2137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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/stable-diffusion-xl-base-1.0",
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."