Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,9 +17,11 @@ translated_questions = {
|
|
17 |
}
|
18 |
|
19 |
# Загружаем русскую модель
|
20 |
-
model_name = "
|
21 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
22 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
|
|
|
|
23 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
24 |
model = model.to(device)
|
25 |
|
|
|
17 |
}
|
18 |
|
19 |
# Загружаем русскую модель
|
20 |
+
model_name = "DeepPavlov/rubert-base-cased"
|
21 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
22 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
23 |
+
|
24 |
+
# Проверка доступности GPU
|
25 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
26 |
model = model.to(device)
|
27 |
|