duchaba commited on
Commit
7d3dadc
·
verified ·
1 Parent(s): 449405a

Update app.py

Browse files

fix v1.get with error

Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -844,6 +844,10 @@ def _censor_me(self, p, safer=0.0005):
844
  resp_dict = resp_orig.model_dump()
845
  #
846
  v1 = resp_dict["results"][0]["category_scores"]
 
 
 
 
847
  max_key = max(v1, key=v1.get)
848
  max_value = v1[max_key]
849
  sum_value = sum(v1.values())
 
844
  resp_dict = resp_orig.model_dump()
845
  #
846
  v1 = resp_dict["results"][0]["category_scores"]
847
+ try:
848
+ keym = vi.get
849
+ except Exception as e:
850
+ keym = 0
851
  max_key = max(v1, key=v1.get)
852
  max_value = v1[max_key]
853
  sum_value = sum(v1.values())