linoyts HF Staff commited on
Commit
07b4ebe
·
verified ·
1 Parent(s): b73a917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -181,7 +181,7 @@ def polish_prompt_hf(original_prompt, img_list):
181
  result = completion.choices[0].message.content
182
 
183
  # Try to extract JSON if present
184
- if '{"Rewritten"' in result:
185
  try:
186
  # Clean up the response
187
  result = result.replace('```json', '').replace('```', '')
 
181
  result = completion.choices[0].message.content
182
 
183
  # Try to extract JSON if present
184
+ if '"Rewritten"' in result:
185
  try:
186
  # Clean up the response
187
  result = result.replace('```json', '').replace('```', '')