Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
25 |
-
output = (model_functions.get(idx) or model_functions.get(1))(prompt,
|
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 |
|