Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

Bert-base-uncased for Android-Ios Question Classification

Code: See Ainize Workspace
Android-Ios-Classification DEMO: Ainize Endpoint
Demo web Code: Github
Android-Ios-Classification API: Ainize API

Overview

Language model: bert-base-cased
Language: English
Training data: Question classification Android-Ios dataset from Kaggle

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

model_path = "EasthShin/Android_Ios_Classification"

tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)

classifier = pipeline('text-classification', model=model_path, tokenizer=tokenizer)

question = "I bought goodnote in Appstore"

result = dict()
result[0] = classifier(question)[0]
Downloads last month
27

Space using EasthShin/Android_Ios_Classification 1