Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import json | |
| from transformers import pipeline | |
| classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli") | |
| def tagging( | |
| text_input: str, | |
| tags: str, | |
| multi_label: bool | |
| ): | |
| try: | |
| candidate_labels = json.loads(tags) | |
| except error: | |
| raise gr.Error(f"Error {error}") | |
| output = classifier(text_input, candidate_labels, multi_label=True) | |
| return output | |
| app = gr.Interface(fn=tagging, inputs=[ | |
| gr.Textbox(label="Input here"), | |
| gr.Textbox(label="Mags here"), | |
| gr.Checkbox(label="Multi_label", value=True) | |
| ], outputs="text") | |
| app.launch() |