Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| model = pipeline("zero-shot-classification", model="facebook/bart-large-mnli") | |
| def predict(field_name, property_list): | |
| properties = property_list.split(",") | |
| results = model(field_name, properties) | |
| top_results = [] | |
| for i, result in enumerate(results["labels"]): | |
| if i < 10: | |
| top_results.append(f"{result}: {results['scores'][i]}") | |
| return results["labels"][0], "\n".join(top_results) | |
| demo = gr.Interface( | |
| fn=predict, inputs=["text", "textarea"], outputs=["text", "textarea"] | |
| ) | |
| demo.launch() | |