import gradio as gr from transformers import pipeline # Load the model model_name = "knowledgator/comprehend_it-base" classifier = pipeline("zero-shot-classification", model=model_name, device="cpu") # Function to classify feedback def classify_feedback(feedback_text): # Classify feedback using the loaded model labels = ["Procedures", "Maintenance", "Operations", "Health", "Safety"] result = classifier(feedback_text, labels, multi_label=True) # Get the top two labels associated with the feedback top_labels = result["labels"][:2] scores = result["scores"][:2] # Generate HTML content for displaying the scores as meters/progress bars html_content = "" for i in range(len(top_labels)): score_percentage = scores[i] * 100 # Convert score to percentage html_content += f"