Spaces:
Sleeping
Sleeping
app.py
CHANGED
|
@@ -12,10 +12,10 @@ databricks_token = os.getenv('TENATCH_TOKEN')
|
|
| 12 |
model_uri = "http://15.152.197.215/v1/completions"
|
| 13 |
|
| 14 |
def extract_json(gen_text, n_shot_learning=0):
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
if(n_shot_learning > 0) :
|
| 20 |
for i in range(0, n_shot_learning):
|
| 21 |
gen_text = gen_text[start_index:]
|
|
|
|
| 12 |
model_uri = "http://15.152.197.215/v1/completions"
|
| 13 |
|
| 14 |
def extract_json(gen_text, n_shot_learning=0):
|
| 15 |
+
if(n_shot_learning == -1) :
|
| 16 |
+
start_index = 0
|
| 17 |
+
else :
|
| 18 |
+
start_index = gen_text.index("### Response:\n{") + 14
|
| 19 |
if(n_shot_learning > 0) :
|
| 20 |
for i in range(0, n_shot_learning):
|
| 21 |
gen_text = gen_text[start_index:]
|