Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,11 +9,9 @@ from transformers import pipeline, AutoTokenizer, TFGPT2LMHeadModel
|
|
9 |
|
10 |
|
11 |
# function to run
|
12 |
-
def run_model(input_text,
|
13 |
-
max_length,
|
14 |
-
length_penalty):
|
15 |
|
16 |
-
#
|
17 |
model_gpt2 = TFGPT2LMHeadModel.from_pretrained("Sultannn/gpt2-ft-id-puisi")
|
18 |
token_gpt2 = AutoTokenizer.from_pretrained("Sultannn/gpt2-ft-id-puisi")
|
19 |
|
@@ -67,7 +65,7 @@ title = "indonesia Puisi Generator"
|
|
67 |
description = "Demo for Puisi Generator ID. Models are GPT-2"
|
68 |
|
69 |
#footer
|
70 |
-
article = "<p style='text-align: center'><a href='https://github.com/sultanbst123/
|
71 |
|
72 |
#run gradio
|
73 |
gr.Interface(
|
@@ -88,15 +86,7 @@ gr.Interface(
|
|
88 |
step=5,
|
89 |
default=100,
|
90 |
label="Max Length(panjang maksimum urutan)",
|
91 |
-
),
|
92 |
-
#length_penalty
|
93 |
-
gr.inputs.Slider(
|
94 |
-
minimum=1,
|
95 |
-
maximum=3,
|
96 |
-
step=1,
|
97 |
-
default=1,
|
98 |
-
label="Length Penalty(pilih 1 jika ingin yang panjang)",
|
99 |
-
),
|
100 |
],
|
101 |
#output text
|
102 |
outputs=
|
|
|
9 |
|
10 |
|
11 |
# function to run
|
12 |
+
def run_model(input_text, max_length):
|
|
|
|
|
13 |
|
14 |
+
#GPT2 FT Transformer
|
15 |
model_gpt2 = TFGPT2LMHeadModel.from_pretrained("Sultannn/gpt2-ft-id-puisi")
|
16 |
token_gpt2 = AutoTokenizer.from_pretrained("Sultannn/gpt2-ft-id-puisi")
|
17 |
|
|
|
65 |
description = "Demo for Puisi Generator ID. Models are GPT-2"
|
66 |
|
67 |
#footer
|
68 |
+
article = "<p style='text-align: center'><a href='https://github.com/sultanbst123/Hugging-Face-indo' target='_blank'><u>Untuk penjelasan lihat di repo ku</u> 😁</a></p>"
|
69 |
|
70 |
#run gradio
|
71 |
gr.Interface(
|
|
|
86 |
step=5,
|
87 |
default=100,
|
88 |
label="Max Length(panjang maksimum urutan)",
|
89 |
+
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
90 |
],
|
91 |
#output text
|
92 |
outputs=
|