Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,13 @@ generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-frau
|
|
7 |
def generate(text):
|
8 |
seed = random.randint(10, 500)
|
9 |
set_seed(seed)
|
10 |
-
result = generator(text,
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
return result[0]["generated_text"]
|
12 |
|
13 |
examples = [
|
|
|
7 |
def generate(text):
|
8 |
seed = random.randint(10, 500)
|
9 |
set_seed(seed)
|
10 |
+
result = generator(text,
|
11 |
+
max_length=500,
|
12 |
+
num_return_sequences=1,
|
13 |
+
do_sample=True,
|
14 |
+
temperature=0.75,
|
15 |
+
top_k=50,
|
16 |
+
top_p=0.9)
|
17 |
return result[0]["generated_text"]
|
18 |
|
19 |
examples = [
|