Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-from model import Model
+from model import Detector
+from PIL import Image
 
+m = Detector()
+m.load('./toxic_detection_res50.gz.tar')
 
-m = Model()
-
+def image_toxic_detect(im):
+    return m.detect(Image.fromarray(im))["toxic_score"]
 
 with gr.Blocks() as app:
     gr.Markdown("app")
@@ -13,7 +16,7 @@ with gr.Blocks() as app:
         image_button_toxic = gr.Button("image_toxic")
         
     # image 
-    image_button_toxic.click(m.image_toxic_detect, inputs=image_input_toxic, outputs=image_output_toxic)
+    image_button_toxic.click(image_toxic_detect, inputs=image_input_toxic, outputs=image_output_toxic)
 
     
 demo.launch(server_name="0.0.0.0")