Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-from model import Detector
 from PIL import Image
+from toxic_detection import TextToxicDetector
+from toxic_detection import ImgToxicDetector
 
-m = Detector()
-m.load('./toxic_detection_res50.gz.tar')
+text_model = TextToxicDetector()
+text_model.load('./toxic_detection_res50.gz.tar')
+img_model = ImgToxicDetector()
+img_model.load('./toxic_detection_res50.gz.tar')
 
 def image_toxic_detect(im):
     return m.detect(Image.fromarray(im))["toxic_score"]