Spaces:
Sleeping
Sleeping
Log tamponlamasını kapat
Browse files
app.py
CHANGED
|
@@ -15,6 +15,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 15 |
|
| 16 |
MODEL_ADI = "melikegks/turkish-pii-guard-0.8b"
|
| 17 |
|
|
|
|
|
|
|
|
|
|
| 18 |
GECERLI_ETIKETLER = {
|
| 19 |
"[AD]", "[TCKN]", "[DOGUM_TARIHI]", "[DOGUM_YERI]", "[ANNE_ADI]",
|
| 20 |
"[ANNE_KIZLIK]", "[BABA_ADI]", "[PASAPORT_NO]", "[EHLIYET_NO]", "[SGK_NO]",
|
|
@@ -64,7 +67,7 @@ except ImportError:
|
|
| 64 |
if not ZEROGPU:
|
| 65 |
torch.set_num_threads(os.cpu_count() or 2)
|
| 66 |
|
| 67 |
-
print("Model yükleniyor… ZeroGPU:", ZEROGPU)
|
| 68 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ADI)
|
| 69 |
model = AutoModelForCausalLM.from_pretrained(
|
| 70 |
MODEL_ADI,
|
|
@@ -74,7 +77,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 74 |
if ZEROGPU or torch.cuda.is_available():
|
| 75 |
model = model.to("cuda")
|
| 76 |
IM_END_ID = tokenizer.convert_tokens_to_ids(IM_END)
|
| 77 |
-
print("Model hazır. Cihaz:", next(model.parameters()).device)
|
| 78 |
|
| 79 |
|
| 80 |
def gpu_gerekiyorsa(fn):
|
|
|
|
| 15 |
|
| 16 |
MODEL_ADI = "melikegks/turkish-pii-guard-0.8b"
|
| 17 |
|
| 18 |
+
# Konteynerde stdout tamponlanır; loglar anında görünsün diye kapatıyoruz.
|
| 19 |
+
os.environ.setdefault("PYTHONUNBUFFERED", "1")
|
| 20 |
+
|
| 21 |
GECERLI_ETIKETLER = {
|
| 22 |
"[AD]", "[TCKN]", "[DOGUM_TARIHI]", "[DOGUM_YERI]", "[ANNE_ADI]",
|
| 23 |
"[ANNE_KIZLIK]", "[BABA_ADI]", "[PASAPORT_NO]", "[EHLIYET_NO]", "[SGK_NO]",
|
|
|
|
| 67 |
if not ZEROGPU:
|
| 68 |
torch.set_num_threads(os.cpu_count() or 2)
|
| 69 |
|
| 70 |
+
print("Model yükleniyor… ZeroGPU:", ZEROGPU, flush=True)
|
| 71 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ADI)
|
| 72 |
model = AutoModelForCausalLM.from_pretrained(
|
| 73 |
MODEL_ADI,
|
|
|
|
| 77 |
if ZEROGPU or torch.cuda.is_available():
|
| 78 |
model = model.to("cuda")
|
| 79 |
IM_END_ID = tokenizer.convert_tokens_to_ids(IM_END)
|
| 80 |
+
print("Model hazır. Cihaz:", next(model.parameters()).device, flush=True)
|
| 81 |
|
| 82 |
|
| 83 |
def gpu_gerekiyorsa(fn):
|