Views
No views yet
dbmdz/bert-base-turkish-cased1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_path = "Kahsi13/tomato-disease-bert"
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForSequenceClassification.from_pretrained(model_path)
7
8text = "Yapraklarda kahverengi lekeler görülüyor"
9inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
10
11with torch.no_grad():
12 outputs = model(**inputs)
13 predicted_class = torch.argmax(outputs.logits, dim=1).item()
14
15# The predicted_class index can be converted to label using label_encoder.pkl (optional)