Spaces:
Runtime error
Runtime error
| from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
| tokenizer = AutoTokenizer.from_pretrained("Imran1/sentimen_analysis_yelp") | |
| model = AutoModelForSequenceClassification.from_pretrained("Imran1/sentimen_analysis_yelp") | |
| from transformers import pipeline | |
| Data= pipeline("text-classification", model=model, tokenizer=tokenizer,top_k=5) | |
| Label=[] | |
| Score=[] | |
| def sentiment(text): | |
| data = Data(text)[0] | |
| for i in range (5): | |
| L=data[i]["label"] | |
| S=data[i]["score"] | |
| Label.append(L) | |
| Score.append(S) | |
| return dict(zip(Label,Score)) | |
| import gradio as gr | |
| exmp=["the food is not good.","oh I really love this food "] | |
| gr.Interface(fn=sentiment, inputs="text", outputs="label", examples=exmp,title= "Yelp reviews").launch() |