Spaces:
Sleeping
Sleeping
| import numpy as np | |
| import os | |
| import gradio as gr | |
| os.environ["WANDB_DISABLED"] = "true" | |
| from datasets import load_dataset, load_metric | |
| from transformers import ( | |
| AutoConfig, | |
| # AutoModelForSequenceClassification, | |
| AutoTokenizer, | |
| TrainingArguments, | |
| logging, | |
| pipeline | |
| ) | |
| analyzer = pipeline( | |
| "sentiment-analysis", model="FFZG-cleopatra/M2SA-text-only" | |
| ) | |
| def predict_sentiment(x): | |
| print(analyzer(x)) | |
| return analyzer(x)[0]["label"] | |
| interface = gr.Interface( | |
| fn=predict_sentiment, | |
| inputs='text', | |
| outputs=['text'], | |
| title='Multilingual Unimodal Sentiment Analysis', | |
| examples= ["I love tea","I hate coffee"], | |
| description='Get the positive/neutral/negative sentiment for the given input.' | |
| ) | |
| interface.launch(inline = False) | |