Spaces:
Running
Running
| # https://huggingface.co/spaces/CK42/sentiment-model-comparison/blob/main/app.py | |
| # import sklearn | |
| from os import O_ACCMODE | |
| import gradio as gr | |
| import joblib | |
| from transformers import pipeline | |
| import requests.exceptions | |
| from huggingface_hub import HfApi, hf_hub_download | |
| from huggingface_hub.repocard import metadata_load | |
| app = gr.Blocks() | |
| model_id_1 = "juliensimon/distilbert-amazon-shoe-reviews" | |
| model_id_2 = "juliensimon/distilbert-amazon-shoe-reviews" | |
| def load_agent(model_id): | |
| """ | |
| This function load the agent's results | |
| """ | |
| # Load the metrics | |
| metadata = get_metadata(model_id) | |
| # get predictions | |
| predictions = predict(model_id) | |
| return model_id, predictions | |
| def get_metadata(model_id): | |
| """ | |
| Get the metadata of the model repo | |
| :param model_id: | |
| :return: metadata | |
| """ | |
| try: | |
| readme_path = hf_hub_download(model_id, filename="README.md") | |
| metadata = metadata_load(readme_path) | |
| print(metadata) | |
| return metadata | |
| except requests.exceptions.HTTPError: | |
| return None | |
| # classifier = pipeline("text-classification", model="juliensimon/distilbert-amazon-shoe-reviews") | |
| def predict(review, model_id): | |
| classifier = pipeline("text-classification", model=model_id) | |
| prediction = classifier(review) | |
| print(prediction) | |
| stars = prediction[0]['label'] | |
| stars = (int)(stars.split('_')[1])+1 | |
| score = 100*prediction[0]['score'] | |
| return "{} {:.0f}%".format("\U00002B50"*stars, score) | |
| with app: | |
| gr.Markdown( | |
| """ | |
| # Compare Sentiment Analysis Models | |
| Type text to predict sentiment. | |
| """) | |
| with gr.Row(): | |
| inp_1= gr.Textbox(label="Type text here.",placeholder="The customer service was satisfactory.") | |
| out_2 = gr.Textbox(label="Prediction") | |
| # gr.Markdown( | |
| # """ | |
| # Model Predictions | |
| # """) | |
| with gr.Row(): | |
| model1_input = gr.Textbox(label="Model 1") | |
| with gr.Row(): | |
| btn = gr.Button("Prediction for Model 1") | |
| btn.click(fn=load_agent(model_id_1), inputs=inp_1, outputs=out_2) | |
| with gr.Row(): | |
| model2_input = gr.Textbox(label="Model 2") | |
| with gr.Row(): | |
| btn = gr.Button("Prediction for Model 2") | |
| btn.click(fn=predict(model_id_2), inputs=inp_1, outputs=out_2) | |
| # app_button.click(load_agent, inputs=[model1_input, model2_input], outputs=[model1_name, model1_score_output, model2_name, model2_score_output]) | |
| # examples = gr.Examples(examples=[["juliensimon/distilbert-amazon-shoe-reviews","juliensimon/distilbert-amazon-shoe-reviews"]], | |
| # inputs=[model1_input, model2_input]) | |
| app.launch() | |