Pamudu13 commited on
Commit
a7d308b
·
verified ·
1 Parent(s): 3a46525

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -2,6 +2,8 @@ import gradio as gr
2
  import time
3
  import random
4
 
 
 
5
  models = [
6
  "Yntec/ChickFlick",
7
  "Yntec/CrystalReality",
@@ -21,8 +23,8 @@ for model_path in models:
21
 
22
  def send_it_idx(idx):
23
  def send_it_fn(prompt):
24
- seed = random.randint(0, 2**32 - 1)
25
- output = (model_functions.get(idx) or model_functions.get(1))(prompt, seed)
26
  return output
27
  return send_it_fn
28
 
@@ -44,6 +46,8 @@ def all_task_end(cnt, t_stamp):
44
  return d, tog
45
 
46
  def all_task_start():
 
 
47
  t_stamp = time.time()
48
  return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
49
 
 
2
  import time
3
  import random
4
 
5
+ random_seed = None
6
+
7
  models = [
8
  "Yntec/ChickFlick",
9
  "Yntec/CrystalReality",
 
23
 
24
  def send_it_idx(idx):
25
  def send_it_fn(prompt):
26
+ global random_seed
27
+ output = (model_functions.get(idx) or model_functions.get(1))(prompt, random_seed)
28
  return output
29
  return send_it_fn
30
 
 
46
  return d, tog
47
 
48
  def all_task_start():
49
+ global random_seed
50
+ random_seed = random.randint(0, 2**32 - 1)
51
  t_stamp = time.time()
52
  return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
53