Update app.py

app.py

@@ -2,11 +2,14 @@ import gradio as gr
 from transformers import pipeline
 
 # 加载 CodeGen 模型
-code_generator = pipeline('text-generation', model='Salesforce/cloudops_tsf')
+code_generator = pipeline('text-generation', model='Salesforce/codegen-350M-multi')  # 确保模型名称正确
 
 def generate_code(prompt):
-    return code_generator(prompt, max_length=10000000)[0]['generated_text']
+    # 设置合理的 max_length
+    result = code_generator(prompt, max_length=150, truncation=True)  # 使用合理的 max_length
+    return result[0]['generated_text']
 
 # 创建 Gradio 接口
-iface = gr.Interface(fn=generate_code, inputs="text", outputs="text")
-iface.launch()
+iface = gr.Interface(fn=generate_code, inputs="text", outputs="text", title="Code Generator",
+                     description="输入您的代码提示，生成代码片段。")
+iface.launch(share=True)  # 设置为 True 以创建公共链接