Upload Detoxify model

app.py

@@ -1,4 +1,7 @@
 import gradio as gr
+<<<<<<< HEAD
+from detoxify import Detoxify
+=======
 import torch.nn as nn
 import torch
 from transformers import BertTokenizerFast as BertTokenizer, BertModel
@@ -103,6 +106,7 @@ model = ToxicCommentTagger.load_from_checkpoint(
 model.eval()
 model.freeze()
 
+>>>>>>> 2a04af3d9d5ddbaa3eb1631c0e56d215462a7e36
 
 all_categories = {'all_categories': [
                                     'toxicity',
@@ -128,8 +132,11 @@ examples = [
 ]
 
 
+model = Detoxify('multilingual')
+
+
 def toxicity(sentence, threshold):
-    predicts = predict(model, tokenizer, sentence)
+    predicts = model.predict(sentence)
     return [ x for x in predicts if predicts[x] > threshold/100 ], all_categories
 
 gr.Interface(fn=toxicity, 
@@ -142,5 +149,3 @@ gr.Interface(fn=toxicity,
                  gr.JSON(all_categories)
              ],
              examples=examples).launch()
-
-

requirements.txt

@@ -1,2 +1,3 @@
-transformers==4.23.1
-pytorch-lightning==1.7.7
+detoxify==0.5.0
+
+