app.py

import gradio as gr
from PIL import Image
from toxic_detection import TextToxicDetector
from toxic_detection import ImgToxicDetector

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"]

with gr.Blocks() as app:
    gr.Markdown("app")
    with gr.Tab("Toxic Image Detector"):
        image_input_toxic = gr.Image()
        image_output_toxic = gr.Label(num_top_classes=2)
        image_button_toxic = gr.Button("image_toxic")
        
    # image 
    image_button_toxic.click(image_toxic_detect, inputs=image_input_toxic, outputs=image_output_toxic)

    
app.launch(server_name="0.0.0.0")