app.py
CHANGED
|
@@ -65,7 +65,15 @@ def extract_json(gen_text, n_shot_learning=0):
|
|
| 65 |
for i in range(0, n_shot_learning):
|
| 66 |
gen_text = gen_text[start_index:]
|
| 67 |
start_index = gen_text.index("### Response:\n{") + 14
|
| 68 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
return gen_text[start_index:end_index]
|
| 70 |
|
| 71 |
def score_model(model_uri, databricks_token, prompt):
|
|
|
|
| 65 |
for i in range(0, n_shot_learning):
|
| 66 |
gen_text = gen_text[start_index:]
|
| 67 |
start_index = gen_text.index("### Response:\n{") + 14
|
| 68 |
+
|
| 69 |
+
end_index = gen_text.find("}\n\n### ")
|
| 70 |
+
if(end_index != -1) :
|
| 71 |
+
print("Start***No end index!!!")
|
| 72 |
+
print(gen_text)
|
| 73 |
+
print("End***No end index!!!")
|
| 74 |
+
|
| 75 |
+
end_index = end_index + 1
|
| 76 |
+
|
| 77 |
return gen_text[start_index:end_index]
|
| 78 |
|
| 79 |
def score_model(model_uri, databricks_token, prompt):
|