Spaces:

debisoft
/

nicolemind

Sleeping

debisoft commited on Dec 11, 2023

Commit

5526a2a

1 Parent(s): c381f62

find

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def extract_json(gen_text, n_shot_learning=0):
         for i in range(0, n_shot_learning):
             gen_text = gen_text[start_index:]
             start_index = gen_text.index("### Response:\n{") + 14
-    end_index = gen_text.index("}\n\n### ") + 1
     return gen_text[start_index:end_index]
 def score_model(model_uri, databricks_token, prompt):

         for i in range(0, n_shot_learning):
             gen_text = gen_text[start_index:]
             start_index = gen_text.index("### Response:\n{") + 14
+    end_index = gen_text.find("}\n\n### ") + 1
     return gen_text[start_index:end_index]
 def score_model(model_uri, databricks_token, prompt):