Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import tensorflow as tf | |
| import tensorflow_text as text | |
| from huggingface_hub import from_pretrained_keras | |
| model = from_pretrained_keras("weightedhuman/fine-tuned-bert-news-classifier") | |
| def get_sentiment_score(text): | |
| if text is not None: | |
| serving_results = model \ | |
| .signatures['serving_default'](tf.constant(text)) | |
| serving_results = tf.sigmoid(serving_results['classifier']) | |
| serving_results_np = serving_results.numpy() | |
| for i in range(len(serving_results_np)): | |
| output_value = serving_results_np[i][0] | |
| return float(output_value) | |
| else: | |
| return "" | |
| intf = gr.Interface( | |
| fn = get_sentiment_score, | |
| inputs = gr.Textbox(), | |
| outputs = gr.Label() | |
| ) | |
| intf.launch() |