Uthar commited on
Commit
54550f2
·
verified ·
1 Parent(s): cd93cdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -69,8 +69,14 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
69
  if width > 0: kwargs["width"] = width
70
  if steps > 0: kwargs["num_inference_steps"] = steps
71
  if cfg > 0: cfg = kwargs["guidance_scale"] = cfg
72
- if seed == -1: kwargs["seed"] = randomize_seed()
73
- else: kwargs["seed"] = seed
 
 
 
 
 
 
74
  task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
75
  await asyncio.sleep(0)
76
  try:
@@ -91,7 +97,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
91
  # png_path = "img.png"
92
  # png_path = get_current_time() + "_" + model_str.replace("/", "_") + ".png"
93
  # png_path = model_str.replace("/", "_") + " - " + prompt + " - " + get_current_time() + ".png"
94
- png_path = model_str.replace("/", "_") + " - " + get_current_time() + ".png"
95
  image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
96
  return image
97
  return None
@@ -129,7 +135,6 @@ CSS="""
129
  js_func = """
130
  function refresh() {
131
  const url = new URL(window.location);
132
-
133
  if (url.searchParams.get('__theme') !== 'dark') {
134
  url.searchParams.set('__theme', 'dark');
135
  window.location.href = url.href;
@@ -138,7 +143,6 @@ function refresh() {
138
  """
139
 
140
  js_AutoSave="""
141
-
142
  console.log("Yo");
143
 
144
  var img1 = document.querySelector("div#component-355 .svelte-1kpcxni button.svelte-1kpcxni .svelte-1kpcxni img"),
@@ -258,4 +262,4 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
258
 
259
  # demo.queue(default_concurrency_limit=200, max_size=200)
260
  demo.launch(show_api=False, max_threads=400)
261
- # demo.launch(show_api=False, max_threads=400, js=js_AutoSave)
 
69
  if width > 0: kwargs["width"] = width
70
  if steps > 0: kwargs["num_inference_steps"] = steps
71
  if cfg > 0: cfg = kwargs["guidance_scale"] = cfg
72
+
73
+ if seed == -1:
74
+ theSeed = randomize_seed()
75
+ kwargs["seed"] = theSeed
76
+ else:
77
+ kwargs["seed"] = seed
78
+ theSeed = seed
79
+
80
  task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
81
  await asyncio.sleep(0)
82
  try:
 
97
  # png_path = "img.png"
98
  # png_path = get_current_time() + "_" + model_str.replace("/", "_") + ".png"
99
  # png_path = model_str.replace("/", "_") + " - " + prompt + " - " + get_current_time() + ".png"
100
+ png_path = model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
101
  image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
102
  return image
103
  return None
 
135
  js_func = """
136
  function refresh() {
137
  const url = new URL(window.location);
 
138
  if (url.searchParams.get('__theme') !== 'dark') {
139
  url.searchParams.set('__theme', 'dark');
140
  window.location.href = url.href;
 
143
  """
144
 
145
  js_AutoSave="""
 
146
  console.log("Yo");
147
 
148
  var img1 = document.querySelector("div#component-355 .svelte-1kpcxni button.svelte-1kpcxni .svelte-1kpcxni img"),
 
262
 
263
  # demo.queue(default_concurrency_limit=200, max_size=200)
264
  demo.launch(show_api=False, max_threads=400)
265
+ # demo.launch(show_api=False, max_threads=400, js=js_AutoSave)