Spaces:

MINDUS-inc
/

LineChatBot-huggingface

Paused

mindus-tk commited on Jan 16, 2024

Commit

7ca8708

1 Parent(s): 72c6fbd

app.py

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("/app.py", methods=['POST'])
-def chat():
     input_data = request.json
     input_text = input_data['input']

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