Spaces:
Running
Running
Add application
Browse files
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)
|