Sultannn commited on
Commit
ae02e13
·
1 Parent(s): af694fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -14
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
- #MBART Transformer
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/Text_summarization-id2id' target='_blank'><u>Untuk penjelasan lihat di repo ku</u> 😁</a></p>"
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=