wuhp commited on
Commit
766e4d2
·
verified ·
1 Parent(s): edd3af7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -137,8 +137,12 @@ def compare_models(models: List[str], img: Image.Image, threshold: float, custom
137
  except Exception as e:
138
  # show blank slate if model unavailable
139
  results.append(Image.new("RGB", img.size, (40, 40, 40)))
140
- err = str(e).split("\n")[0][:120] # shorten
141
- legends[m] = f"ERROR: {err}"
 
 
 
 
142
 
143
  return results, legends
144
 
 
137
  except Exception as e:
138
  # show blank slate if model unavailable
139
  results.append(Image.new("RGB", img.size, (40, 40, 40)))
140
+ err_msg = str(e)
141
+ # Normalize common weight‑missing errors for clarity
142
+ if "No such file or directory" in err_msg:
143
+ legends[m] = "Unavailable (weights not found)"
144
+ else:
145
+ legends[m] = f"ERROR: {err_msg.splitlines()[0][:120]}"
146
 
147
  return results, legends
148