Spaces:
Sleeping
Sleeping
kauabarros-24
commited on
Commit
·
03e3385
1
Parent(s):
aa342c7
CHORE: Added potas_recommend
Browse files
potas_recommend → src/recommend/potas_recommend
RENAMED
|
File without changes
|
src/recommend/routes/ai_routes.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch
|
|
| 3 |
from fastapi import APIRouter
|
| 4 |
from transformers import BertForSequenceClassification, BertTokenizer
|
| 5 |
|
| 6 |
-
model_name = "potas_recommend"
|
| 7 |
model = BertForSequenceClassification.from_pretrained(model_name)
|
| 8 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
| 9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 3 |
from fastapi import APIRouter
|
| 4 |
from transformers import BertForSequenceClassification, BertTokenizer
|
| 5 |
|
| 6 |
+
model_name = "src/recommend/potas_recommend"
|
| 7 |
model = BertForSequenceClassification.from_pretrained(model_name)
|
| 8 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
| 9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|