onuralpszr commited on
Commit
66c6695
Β·
verified Β·
1 Parent(s): 8183bdd

fix: 🐞 remove extra text split

Browse files

Signed-off-by: Onuralp SEZER <[email protected]>

Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -63,9 +63,7 @@ def run_image(
63
  runner = load_runner()
64
  with open("input.jpeg", "wb") as f:
65
  f.write(input_image)
66
-
67
- class_names = [class_name.strip() for class_name in class_names.split(',')]
68
-
69
  texts = [[t.strip()] for t in class_names.split(",")] + [[" "]]
70
  data_info = runner.pipeline(dict(img_id=0, img_path="input.jpeg",
71
  texts=texts))
 
63
  runner = load_runner()
64
  with open("input.jpeg", "wb") as f:
65
  f.write(input_image)
66
+
 
 
67
  texts = [[t.strip()] for t in class_names.split(",")] + [[" "]]
68
  data_info = runner.pipeline(dict(img_id=0, img_path="input.jpeg",
69
  texts=texts))