TomokiFujihara commited on
Commit
8c2b3e8
1 Parent(s): ab06164

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def generate(text):
16
  outputs = model(inputs['input_ids'], inputs['attention_mask']).detach().numpy()[:3]
17
  # 銈广偝銈€倰绠楀嚭
18
  minimum = np.min(outputs)
19
- if min < 0:
20
  outputs = outputs - minimum
21
  score = outputs / np.sum(outputs)
22
  prediction = f'Not Offensive: {outputs[0]}, Gray-area: {outputs[1]}, Offensive: {outputs[1]}'
 
16
  outputs = model(inputs['input_ids'], inputs['attention_mask']).detach().numpy()[:3]
17
  # 銈广偝銈€倰绠楀嚭
18
  minimum = np.min(outputs)
19
+ if minimum < 0:
20
  outputs = outputs - minimum
21
  score = outputs / np.sum(outputs)
22
  prediction = f'Not Offensive: {outputs[0]}, Gray-area: {outputs[1]}, Offensive: {outputs[1]}'