debisoft commited on
Commit
9c57288
1 Parent(s): c60f222
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def extract_json(gen_text, n_shot_learning=0):
66
  for i in range(0, n_shot_learning):
67
  gen_text = gen_text[start_index:]
68
  start_index = gen_text.index("### Response:\n{") + 14
69
- end_index = gen_text.find("}\n\n### ") + 1
70
  if(end_index != -1) :
71
  end_index = end_index + 1
72
 
 
66
  for i in range(0, n_shot_learning):
67
  gen_text = gen_text[start_index:]
68
  start_index = gen_text.index("### Response:\n{") + 14
69
+ end_index = gen_text.find("}\n\n### ")
70
  if(end_index != -1) :
71
  end_index = end_index + 1
72