Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import sys
-from generate import generate_main
+from generate import main
 
 def generate_text(length, nsamples, prefix, temperature):
  my_prefix = "--prefix=" + prefix + ","
@@ -13,7 +13,7 @@ def generate_text(length, nsamples, prefix, temperature):
  f'--temperature={temperature}', 
  '--model_path=model/model_epoch10'
  ]
- output = generate_main()
+ output = main()
  return output
 
 input_length = gr.inputs.Number(label="生成文本长度", min_value=1, max_value=10000, default=1000)