Spaces:
Sleeping
Sleeping
Commit
·
5f25ca2
1
Parent(s):
210b4bb
added
Browse files
app.py
CHANGED
@@ -57,7 +57,12 @@ def load_model():
|
|
57 |
return processor, model
|
58 |
|
59 |
def extract_json_from_llm_output(llm_result):
|
60 |
-
|
|
|
|
|
|
|
|
|
|
|
61 |
if match:
|
62 |
return match.group(0)
|
63 |
return None
|
|
|
57 |
return processor, model
|
58 |
|
59 |
def extract_json_from_llm_output(llm_result):
|
60 |
+
# Try to extract JSON from a code block first (```json ... ``` or ``` ... ```)
|
61 |
+
code_block_match = re.search(r"```(?:json)?\s*({[\s\S]*?})\s*```", llm_result, re.IGNORECASE)
|
62 |
+
if code_block_match:
|
63 |
+
return code_block_match.group(1)
|
64 |
+
# Fallback: extract first {...} block
|
65 |
+
match = re.search(r'\{[\s\S]*\}', llm_result)
|
66 |
if match:
|
67 |
return match.group(0)
|
68 |
return None
|