melikegks commited on
Commit
b401778
·
verified ·
1 Parent(s): f9fbcba

Log tamponlamasını kapat

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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):