OsBaran commited on
Commit
c9670f4
·
1 Parent(s): 198afc2

Add application

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipe
4
  # Buraya İngilizce modelinizi yazın
5
  model = AutoModelForSequenceClassification.from_pretrained("OsBaran/Roberta-Classification-Model")
6
  tokenizer = AutoTokenizer.from_pretrained("roberta-base")
7
-
8
  def predict_with_roberta(model, tokenizer, input_text):
9
  # Giriş metnini tokenize et ve tensor'a çevir
10
  inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True).to(device)
 
4
  # Buraya İngilizce modelinizi yazın
5
  model = AutoModelForSequenceClassification.from_pretrained("OsBaran/Roberta-Classification-Model")
6
  tokenizer = AutoTokenizer.from_pretrained("roberta-base")
7
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # Cihazı kontrol et
8
  def predict_with_roberta(model, tokenizer, input_text):
9
  # Giriş metnini tokenize et ve tensor'a çevir
10
  inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True).to(device)