DetGC commited on
Commit
900a4ee
·
verified ·
1 Parent(s): e986833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -6,7 +6,9 @@ import os
6
  from threading import RLock
7
  from datetime import datetime
8
 
9
- preSetPrompt = "a tall slender athletic nude caucasian young woman. perky breasts. gorgeous face. sexy eyes. feme fatale. provocative. stainles steel metal dildo. masturbation. pussy dripping white fluid. high tech space ship interiour."
 
 
10
  negPreSetPrompt = "[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness"
11
 
12
  lock = RLock()
@@ -39,7 +41,6 @@ max_images = 6
39
  inference_timeout = 400
40
  default_models = models[:num_models]
41
  MAX_SEED = 2**32-1
42
- loopcounter = 1
43
 
44
 
45
  def extend_choices(choices):
@@ -85,11 +86,9 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
85
  if task.done() and result is not None and not isinstance(result, tuple):
86
  with lock:
87
  # png_path = "img.png"
88
- global loopcounter
89
- # png_path = get_current_time() + "_" + str(loopcounter) + ".png"
90
- png_path = get_current_time() + "_" + model_str.replace("/", "_") + ".png"
91
- # png_path = get_current_time() + "_" + str(models_load[model_str]) + "_" + str(seed) + ".png"
92
- loopcounter += 1
93
  image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
94
  return image
95
  return None
@@ -220,11 +219,11 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
220
  steps2 = gr.Slider(label="Number of inference steps", info="If 0, the default value is used.", maximum=100, step=1, value=0)
221
  cfg2 = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0)
222
  seed2 = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
223
- seed_rand2 = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
224
  seed_rand2.click(randomize_seed, None, [seed2], queue=False)
225
  num_images = gr.Slider(1, max_images, value=max_images, step=1, label='Number of images')
226
  with gr.Row():
227
- gen_button2 = gr.Button('Generate', variant='primary', scale=2)
228
  #stop_button2 = gr.Button('Stop', variant='stop', interactive=False, scale=1)
229
  #gen_button2.click(lambda: gr.update(interactive=True), None, stop_button2)
230
 
 
6
  from threading import RLock
7
  from datetime import datetime
8
 
9
+
10
+ preSetPrompt = "cute tall slender athletic 20+ caucasian woman. gorgeous face. perky tits. sly smile. lifting shirt. explicit pose. artistic. photorealistic. cinematic. f1.4"
11
+ # preSetPrompt = "cute tall slender athletic 20+ nude caucasian woman. gorgeous face. perky tits. gaping outie pussy. pussy juice. sly smile. explicit pose. artistic. photorealistic. cinematic. f1.4"
12
  negPreSetPrompt = "[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness"
13
 
14
  lock = RLock()
 
41
  inference_timeout = 400
42
  default_models = models[:num_models]
43
  MAX_SEED = 2**32-1
 
44
 
45
 
46
  def extend_choices(choices):
 
86
  if task.done() and result is not None and not isinstance(result, tuple):
87
  with lock:
88
  # png_path = "img.png"
89
+ # png_path = get_current_time() + "_" + model_str.replace("/", "_") + ".png"
90
+ # png_path = model_str.replace("/", "_") + " - " + prompt + " - " + get_current_time() + ".png"
91
+ png_path = model_str.replace("/", "_") + " - " + get_current_time() + ".png"
 
 
92
  image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
93
  return image
94
  return None
 
219
  steps2 = gr.Slider(label="Number of inference steps", info="If 0, the default value is used.", maximum=100, step=1, value=0)
220
  cfg2 = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0)
221
  seed2 = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
222
+ seed_rand2 = gr.Button("Randomize Seed", size="sm", variant="secondary")
223
  seed_rand2.click(randomize_seed, None, [seed2], queue=False)
224
  num_images = gr.Slider(1, max_images, value=max_images, step=1, label='Number of images')
225
  with gr.Row():
226
+ gen_button2 = gr.Button('Let the machine halucinate', variant='primary', scale=2)
227
  #stop_button2 = gr.Button('Stop', variant='stop', interactive=False, scale=1)
228
  #gen_button2.click(lambda: gr.update(interactive=True), None, stop_button2)
229