aifeifei798 commited on
Commit
ec5018f
·
verified ·
1 Parent(s): fc93d1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -17,7 +17,7 @@ def feifeimodload():
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
18
 
19
  pipe = DiffusionPipeline.from_pretrained(
20
- "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
21
  ).to(device)
22
 
23
  pipe.load_lora_weights(
@@ -67,11 +67,11 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
67
  return image, seed
68
 
69
  examples = [
70
- "this photo is jpop a smile girl in white bikini,snowing",
71
  "this photo is jpop a girl in bikini",
72
  "this photo is kpop a girl in bikini",
73
  "this photo is a girl in bikini",
74
- "DarkIdol-flux + feifei-flux-lora",
75
  "this photo is real model slight smile girl in real life",
76
  "this photo is real model smile girl in real life",
77
  "this photo is real model girl in real life",
@@ -94,9 +94,7 @@ css="""
94
  with gr.Blocks(css=css) as demo:
95
 
96
  with gr.Column(elem_id="col-container"):
97
- gr.Markdown(f"""# DarkIdol-flux + feifei-flux-lora
98
- DarkIdol-flux + feifei-flux-lora is a text-to-image AI model designed to create aesthetic, detailed and diverse images from textual prompts in just 6-8 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
99
- """)
100
 
101
  with gr.Row():
102
 
 
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
18
 
19
  pipe = DiffusionPipeline.from_pretrained(
20
+ "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
21
  ).to(device)
22
 
23
  pipe.load_lora_weights(
 
67
  return image, seed
68
 
69
  examples = [
70
+ "this photo is jpop a smile girl in white bikini",
71
  "this photo is jpop a girl in bikini",
72
  "this photo is kpop a girl in bikini",
73
  "this photo is a girl in bikini",
74
+ "feifei-flux-lora",
75
  "this photo is real model slight smile girl in real life",
76
  "this photo is real model smile girl in real life",
77
  "this photo is real model girl in real life",
 
94
  with gr.Blocks(css=css) as demo:
95
 
96
  with gr.Column(elem_id="col-container"):
97
+ gr.Markdown(f"""# FLUX.1 + feifei-flux-lora """)
 
 
98
 
99
  with gr.Row():
100