Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from deep_translator import GoogleTranslator
|
|
| 9 |
|
| 10 |
# Project by Nymbo
|
| 11 |
|
| 12 |
-
API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-
|
| 13 |
API_TOKEN = os.getenv("HF_READ_TOKEN")
|
| 14 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
| 15 |
timeout = 100
|
|
@@ -28,7 +28,7 @@ def convert_to_png(image):
|
|
| 28 |
png_buffer.seek(0)
|
| 29 |
return Image.open(png_buffer)
|
| 30 |
|
| 31 |
-
def query(prompt, is_negative=False, steps=
|
| 32 |
seed=-1, strength=0.7, width=1024, height=1024):
|
| 33 |
if not prompt:
|
| 34 |
return None
|
|
@@ -111,7 +111,7 @@ h1 {
|
|
| 111 |
"""
|
| 112 |
|
| 113 |
with gr.Blocks(theme=gr.themes.Default(primary_hue="green"), css=css) as app:
|
| 114 |
-
gr.HTML("<center><h1>BSP
|
| 115 |
|
| 116 |
with gr.Column(elem_id="app-container"):
|
| 117 |
with gr.Row():
|
|
@@ -134,7 +134,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="green"), css=css) as app:
|
|
| 134 |
width = gr.Slider(1024, label="Width", minimum=512, maximum=2048, step=64)
|
| 135 |
height = gr.Slider(1024, label="Height", minimum=512, maximum=2048, step=64)
|
| 136 |
with gr.Row():
|
| 137 |
-
steps = gr.Slider(
|
| 138 |
cfg = gr.Slider(7.0, label="CFG Scale", minimum=1.0, maximum=20.0, step=0.5)
|
| 139 |
with gr.Row():
|
| 140 |
strength = gr.Slider(0.7, label="Strength", minimum=0.1, maximum=1.0, step=0.01)
|
|
@@ -150,7 +150,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="green"), css=css) as app:
|
|
| 150 |
with gr.Row():
|
| 151 |
output_image = gr.Image(
|
| 152 |
type="pil",
|
| 153 |
-
label="Generated
|
| 154 |
format="png", # Explicitly set output format
|
| 155 |
elem_id="gallery"
|
| 156 |
)
|
|
|
|
| 9 |
|
| 10 |
# Project by Nymbo
|
| 11 |
|
| 12 |
+
API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
|
| 13 |
API_TOKEN = os.getenv("HF_READ_TOKEN")
|
| 14 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
| 15 |
timeout = 100
|
|
|
|
| 28 |
png_buffer.seek(0)
|
| 29 |
return Image.open(png_buffer)
|
| 30 |
|
| 31 |
+
def query(prompt, is_negative=False, steps=4, cfg_scale=7, sampler="DPM++ 2M Karras",
|
| 32 |
seed=-1, strength=0.7, width=1024, height=1024):
|
| 33 |
if not prompt:
|
| 34 |
return None
|
|
|
|
| 111 |
"""
|
| 112 |
|
| 113 |
with gr.Blocks(theme=gr.themes.Default(primary_hue="green"), css=css) as app:
|
| 114 |
+
gr.HTML("<center><h1>BSP Low Work</h1></center>")
|
| 115 |
|
| 116 |
with gr.Column(elem_id="app-container"):
|
| 117 |
with gr.Row():
|
|
|
|
| 134 |
width = gr.Slider(1024, label="Width", minimum=512, maximum=2048, step=64)
|
| 135 |
height = gr.Slider(1024, label="Height", minimum=512, maximum=2048, step=64)
|
| 136 |
with gr.Row():
|
| 137 |
+
steps = gr.Slider(4, label="Steps", minimum=4, maximum=100, step=1)
|
| 138 |
cfg = gr.Slider(7.0, label="CFG Scale", minimum=1.0, maximum=20.0, step=0.5)
|
| 139 |
with gr.Row():
|
| 140 |
strength = gr.Slider(0.7, label="Strength", minimum=0.1, maximum=1.0, step=0.01)
|
|
|
|
| 150 |
with gr.Row():
|
| 151 |
output_image = gr.Image(
|
| 152 |
type="pil",
|
| 153 |
+
label="Generated Image",
|
| 154 |
format="png", # Explicitly set output format
|
| 155 |
elem_id="gallery"
|
| 156 |
)
|