yetessam commited on
Commit
4d703d1
·
verified ·
1 Parent(s): e67267c

Update tools/polite_guard.py

Browse files
Files changed (1) hide show
  1. tools/polite_guard.py +1 -1
tools/polite_guard.py CHANGED
@@ -45,7 +45,7 @@ class PoliteGuardTool(Tool):
45
  classifier = pipeline("text-classification", "Intel/polite-guard")
46
  result = classifier(input_text)
47
  print(f"return {str(result)}")
48
- return result[0]
49
 
50
  except Exception as e:
51
  return f"Error fetching classification for text '{input_text}': {str(e)}"
 
45
  classifier = pipeline("text-classification", "Intel/polite-guard")
46
  result = classifier(input_text)
47
  print(f"return {str(result)}")
48
+ return result
49
 
50
  except Exception as e:
51
  return f"Error fetching classification for text '{input_text}': {str(e)}"