Update app.py
Browse files
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 |
-
"
|
| 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
|
| 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 |
-
"
|
| 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"""#
|
| 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 |
|