Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import tensorflow as tf | |
| from transformers import AutoTokenizer, TFAutoModelForSequenceClassification, TextClassificationPipeline | |
| tokenizer = AutoTokenizer.from_pretrained("dipesh/Intent-Classification-Bert-Base-Cased") | |
| model = TFAutoModelForSequenceClassification.from_pretrained("dipesh/Intent-Classification-Bert-Base-Cased") | |
| intent_classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=False, | |
| framework='tf') | |
| def predict(input_text): | |
| ans = intent_classifier(input_text) | |
| return ans[0]['label'] | |
| predict("hello") | |
| iface = gr.Interface(fn=predict, inputs="text", outputs="text") | |
| iface.launch() | |