Spaces:

MINDUS-inc
/

LineChatBot-huggingface

Paused

mindus-tk commited on Jan 16, 2024

Commit

c7b35c3

1 Parent(s): 5aa4a80

set endpoint

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ model_name = "rinna/youri-7b-chat"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
-@app.route("/", methods=['POST'])
-def infer():
     input_data = request.json
     input_text = input_data['input']

 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
+@app.route("/endpoint", methods=['POST'])
+def endpoint():
     input_data = request.json
     input_text = input_data['input']