fixed code v2

This commit is contained in:
SlimusMinus
2026-08-06 23:45:38 +03:00
parent fdf2e2a8d5
commit 4c32c5f4dd
7 changed files with 598 additions and 7 deletions

View File

@@ -1,4 +1,5 @@
import torch
import os
import torch
from transformers import AutoImageProcessor
from transformers import AutoModelForImageClassification
from transformers import AutoModelForSequenceClassification
@@ -17,6 +18,11 @@ class ModelManager:
self.image_model = None
self.image_processor = None
def _setup_hf_token(self):
if settings.HF_TOKEN:
os.environ["HF_TOKEN"] = settings.HF_TOKEN
os.environ["HUGGING_FACE_HUB_TOKEN"] = settings.HF_TOKEN
def load_device(self):
if settings.DEVICE == "auto":
self.device = torch.device(