avans06 commited on
Commit
c8d7615
·
1 Parent(s): c4123d8

Fixed the JSONDecodeError issue.

Browse files

Use ast.literal_eval to parse text into a tuple in remove_image_from_gallery.

Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +3 -2
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 💬
4
  colorFrom: purple
5
  colorTo: indigo
6
  sdk: gradio
7
- sdk_version: 5.14.0
8
  app_file: app.py
9
  pinned: true
10
  ---
 
4
  colorFrom: purple
5
  colorTo: indigo
6
  sdk: gradio
7
+ sdk_version: 5.16.0
8
  app_file: app.py
9
  pinned: true
10
  ---
app.py CHANGED
@@ -11,6 +11,7 @@ import traceback
11
  import tempfile
12
  import zipfile
13
  import re
 
14
  from datetime import datetime
15
  from collections import defaultdict
16
  from classifyTags import classify_tags
@@ -483,7 +484,7 @@ def get_selection_from_gallery(gallery: list, tag_results: dict, selected_state:
483
  if not selected_state:
484
  return selected_state
485
 
486
- tag_result = { "strings": "", "classified_tags": "", "rating": "", "character_res": "", "general_res": "", "unclassified_tags": "" }
487
  if selected_state.value["image"]["path"] in tag_results:
488
  tag_result = tag_results[selected_state.value["image"]["path"]]
489
 
@@ -515,7 +516,7 @@ def remove_image_from_gallery(gallery: list, selected_image: str):
515
  if not gallery or not selected_image:
516
  return gallery
517
 
518
- selected_image = eval(selected_image)
519
  # Remove the selected image from the gallery
520
  if selected_image in gallery:
521
  gallery.remove(selected_image)
 
11
  import tempfile
12
  import zipfile
13
  import re
14
+ import ast
15
  from datetime import datetime
16
  from collections import defaultdict
17
  from classifyTags import classify_tags
 
484
  if not selected_state:
485
  return selected_state
486
 
487
+ tag_result = { "strings": "", "classified_tags": "{}", "rating": "", "character_res": "", "general_res": "", "unclassified_tags": "{}" }
488
  if selected_state.value["image"]["path"] in tag_results:
489
  tag_result = tag_results[selected_state.value["image"]["path"]]
490
 
 
516
  if not gallery or not selected_image:
517
  return gallery
518
 
519
+ selected_image = ast.literal_eval(selected_image) #Use ast.literal_eval to parse text into a tuple.
520
  # Remove the selected image from the gallery
521
  if selected_image in gallery:
522
  gallery.remove(selected_image)
requirements.txt CHANGED
@@ -4,7 +4,7 @@ pillow>=9.0.0
4
  onnxruntime>=1.12.0
5
  huggingface-hub
6
 
7
- gradio==5.14.0
8
  pandas
9
 
10
  # for reorganize WD Tagger into a readable article by Llama3 model.
 
4
  onnxruntime>=1.12.0
5
  huggingface-hub
6
 
7
+ gradio==5.16.0
8
  pandas
9
 
10
  # for reorganize WD Tagger into a readable article by Llama3 model.