Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,11 @@ def generate(
|
|
22 |
top_p = float(top_p)
|
23 |
seed = float(seed)
|
24 |
|
|
|
|
|
|
|
|
|
|
|
25 |
generate_kwargs = dict(
|
26 |
temperature=temperature,
|
27 |
max_new_tokens=max_new_tokens,
|
@@ -96,7 +101,7 @@ additional_inputs=[
|
|
96 |
label="Seed",
|
97 |
value=42,
|
98 |
minimum=1,
|
99 |
-
info="Use an integer starting point to initiate the generation process",
|
100 |
),
|
101 |
]
|
102 |
|
|
|
22 |
top_p = float(top_p)
|
23 |
seed = float(seed)
|
24 |
|
25 |
+
if seed == 0:
|
26 |
+
seed = random.randint(1, 100000)
|
27 |
+
else:
|
28 |
+
seed = seed
|
29 |
+
|
30 |
generate_kwargs = dict(
|
31 |
temperature=temperature,
|
32 |
max_new_tokens=max_new_tokens,
|
|
|
101 |
label="Seed",
|
102 |
value=42,
|
103 |
minimum=1,
|
104 |
+
info="Use an integer starting point to initiate the generation process, put 0 for a random",
|
105 |
),
|
106 |
]
|
107 |
|