TymaaHammouda commited on
Commit
4027959
·
verified ·
1 Parent(s): e725104

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def predict(request: ConflictDetectionRequest):
49
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
  # print(response.split("### Response:")[-1].strip())
51
 
52
- content = {"resp": response.split("### Response:")[-1].strip(), "statusText": "OK","statusCode" : 0}
53
 
54
  return JSONResponse(
55
  content=content,
 
49
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
  # print(response.split("### Response:")[-1].strip())
51
 
52
+ content = {"resp": response.split("</think>")[1].strip(), "statusText": "OK","statusCode" : 0}
53
 
54
  return JSONResponse(
55
  content=content,