Update app.py

app.py

@@ -1,19 +1,17 @@
 import gradio as gr
-import os
-import sys
-from generate import main
+import subprocess
+
 
 def generate_text(length, nsamples, prefix, temperature):
+ # 构建命令行参数
  my_prefix = "--prefix=" + prefix + ","
- sys.argv = [
- 'generate.py', 
- f'--length={length}', 
- f'--nsamples={nsamples}', 
- my_prefix, 
- f'--temperature={temperature}', 
- '--model_path=model/model_epoch10'
- ]
- output = main()
+ args = ["python", "generate.py", f"--length={length}", f"--nsamples={nsamples}", my_prefix, f"--temperature={temperature}", "--model_path=model/model_epoch10"]
+ 
+ # 执行命令并获取输出
+ process = subprocess.Popen(args, stdout=subprocess.PIPE)
+ output, error = process.communicate()
+ output = output.decode("utf-8")
+ 
  return output
 
 input_length = gr.Number(label="生成文本长度", min_value=1, max_value=10000, default=1000)