neo7team commited on
Commit
fe4eee6
·
verified ·
1 Parent(s): b1f33f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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-dev"
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=20, 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,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 Dev Work</h1></center>")
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(35, label="Steps", minimum=10, 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,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 PNG Image",
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
  )