File size: 439 Bytes
6d3c376 af3e7fb 6d3c376 af3e7fb 6d3c376 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
from transformers import pipeline
# 加载模型
pipe = pipeline("text-generation", model="IndexTeam/Index-1.9B-Character", trust_remote_code=True)
# 定义Gradio接口
def generate_text(prompt):
messages = [
{"role": "user", "content": prompt},
]
result = pipe(messages)
return result[0]['generated_text']
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
iface.launch()
|