Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
import sys
|
4 |
-
from generate import
|
5 |
|
6 |
def generate_text(length, nsamples, prefix, temperature):
|
7 |
my_prefix = "--prefix=" + prefix + ","
|
@@ -13,7 +13,7 @@ def generate_text(length, nsamples, prefix, temperature):
|
|
13 |
f'--temperature={temperature}',
|
14 |
'--model_path=model/model_epoch10'
|
15 |
]
|
16 |
-
output =
|
17 |
return output
|
18 |
|
19 |
input_length = gr.inputs.Number(label="生成文本长度", min_value=1, max_value=10000, default=1000)
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
import sys
|
4 |
+
from generate import main
|
5 |
|
6 |
def generate_text(length, nsamples, prefix, temperature):
|
7 |
my_prefix = "--prefix=" + prefix + ","
|
|
|
13 |
f'--temperature={temperature}',
|
14 |
'--model_path=model/model_epoch10'
|
15 |
]
|
16 |
+
output = main()
|
17 |
return output
|
18 |
|
19 |
input_length = gr.inputs.Number(label="生成文本长度", min_value=1, max_value=10000, default=1000)
|