huduo commited on
Commit
826e173
·
verified ·
1 Parent(s): 283f768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # 加载 CodeGen 模型
5
- code_generator = pipeline('text-generation', model='Salesforce/LLaMA-3-8B-SFR-Iterative-DPO-R')
6
 
7
  def generate_code(prompt):
8
  return code_generator(prompt, max_length=10000000)[0]['generated_text']
 
2
  from transformers import pipeline
3
 
4
  # 加载 CodeGen 模型
5
+ code_generator = pipeline('text-generation', model='Salesforce/cloudops_tsf')
6
 
7
  def generate_code(prompt):
8
  return code_generator(prompt, max_length=10000000)[0]['generated_text']