kkAsmaa commited on
Commit
0bbfcb4
·
verified ·
1 Parent(s): 3032abb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  import re
3
  import os
4
  import torch
5
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
6
 
7
  # استدعاء معالج التنظيف الرسمي لـ AraBERT الخاص بكِ
8
  from arabert.preprocess import ArabertPreprocessor
@@ -14,10 +15,10 @@ MODEL_NAME = "aubmindlab/bert-base-arabertv02-twitter"
14
  # سحب المفتاح السري تلقائياً وبأعلى درجات الأمان السيبراني
15
  HF_TOKEN = os.getenv("HF_TOKEN")
16
 
17
- print("🔄 Initializing ChildShield secure environment with standard stable tokenizer...")
18
 
19
- # 🎯 السطر الذهبي المصلح للخطأ: تحويل use_fast إلى False لإجبار السيرفر على الاستقرار
20
- tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, token=HF_TOKEN, use_fast=False)
21
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO, token=HF_TOKEN)
22
  model.eval()
23
 
@@ -81,7 +82,7 @@ def predict_safety_api(text):
81
  outputs = model(**inputs)
82
 
83
  probs = torch.softmax(outputs.logits, dim=-1).flatten().tolist()
84
- unsafe_p = float(probs)
85
 
86
  if unsafe_p > 0.50:
87
  is_blocked = True
@@ -103,4 +104,3 @@ interface = gr.Interface(
103
 
104
  if __name__ == "__main__":
105
  interface.launch()
106
-
 
2
  import re
3
  import os
4
  import torch
5
+ # 🎯 التعديل الحاسم: استدعاء المترجم الصريح لمعمارية بيرت لكسر التعليق التقني
6
+ from transformers import BertTokenizer, AutoModelForSequenceClassification
7
 
8
  # استدعاء معالج التنظيف الرسمي لـ AraBERT الخاص بكِ
9
  from arabert.preprocess import ArabertPreprocessor
 
15
  # سحب المفتاح السري تلقائياً وبأعلى درجات الأمان السيبراني
16
  HF_TOKEN = os.getenv("HF_TOKEN")
17
 
18
+ print("🔄 Instantiating explicit BertTokenizer to force green deployment...")
19
 
20
+ # 🎯 السطر الملكي المصلح: استخدام BertTokenizer من المستودع الأصلي المفتوح لتفادي تعليق الملف الخاص
21
+ tokenizer = BertTokenizer.from_pretrained(MODEL_NAME)
22
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO, token=HF_TOKEN)
23
  model.eval()
24
 
 
82
  outputs = model(**inputs)
83
 
84
  probs = torch.softmax(outputs.logits, dim=-1).flatten().tolist()
85
+ unsafe_p = float(probs[1]) # 🎯 التأكيد الصارم على قراءة معامل الحظر الفعلي
86
 
87
  if unsafe_p > 0.50:
88
  is_blocked = True
 
104
 
105
  if __name__ == "__main__":
106
  interface.launch()