30 lines
610 B
JSON
30 lines
610 B
JSON
{
|
|
"task": "text-generation",
|
|
"model": {
|
|
"type": "transformers",
|
|
"path": "."
|
|
},
|
|
"inference": {
|
|
"script": "inference.py",
|
|
"function": "inference",
|
|
"parameters": [
|
|
{
|
|
"name": "input_text",
|
|
"type": "string",
|
|
"description": "请输入要处理的文本，例如‘推荐河北的旅游景点’",
|
|
"required": true
|
|
}
|
|
],
|
|
"output": {
|
|
"type": "string",
|
|
"description": "模型生成的推荐文本"
|
|
}
|
|
},
|
|
"dependencies": {
|
|
"requirements": "requirements.txt"
|
|
},
|
|
"resources": {
|
|
"instance_type": "GPU-16GB"
|
|
}
|
|
}
|