Upload 2 files

app.py

@@ -0,0 +1,19 @@
+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)

requirements.txt

@@ -0,0 +1,3 @@
+gradio==3.16.2
+transformers==4.21.2
+torch==1.12.1