from transformers import pipeline
import gradio as gr
import tensorflow

model_name = "facebook/bart-large-mnli"

class_labels = ["Misconduct" , "Negligence" , "Discrimination" , "Corruption" , "Violation of Rights" , "Inefficiency" ,
           "Unprofessional Conduct", "Response Time" , "Use of Firearms" , "Property Damage"]


classifier = pipeline("zero-shot-classification", model=model_name)

def zeroshotclassification(input_text):
    issue = classifier(input_text , class_labels , multi_label = False)['labels'][0]
    return issue

iface = gr.Interface(
    fn=zeroshotclassification,
    inputs=[
       "text" 
    ],
    outputs=["text"]
)
iface.launch()