app.py

import gradio as gr
from transformers import pipeline

# 加载模型和tokenizer
generator = pipeline('text-generation', model='hfl/chinese-llama-2-1.3b')

# 定义预测函数
def predict(text,password,temperature):
    if password != "123":
      return "Password wrong"
    else:
      result = generator(text, max_length=400, num_return_sequences=1 , temperature=float(temperature))
      return result[0]['generated_text']

# 创建Gradio界面
demo = gr.Interface(fn=predict,inputs=[gr.Textbox(label="Text"),gr.Textbox(label="Password"),gr.Textbox(label="temperature")],outputs="text",title="Chinese-llama-2-1.3b",description="input your text")

# 启动Gradio界面
demo.launch(share=True)