MHD011 commited on
Commit
2d66b35
·
verified ·
1 Parent(s): 57bf9b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,7 +13,8 @@ logging.basicConfig(level=logging.INFO)
13
  logger = logging.getLogger(__name__)
14
 
15
  # --- إعداد النموذج ---
16
- MODEL_NAME = "tscholak/sqlcoder" # يمكنك تغييره لنموذج آخر إذا رغبت
 
17
 
18
  tokenizer = None
19
  model = None
@@ -24,7 +25,7 @@ def initialize():
24
  logger.info(f"تحميل النموذج على الجهاز: {device}")
25
 
26
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
27
- model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
28
  logger.info("تم تحميل النموذج بنجاح")
29
 
30
  initialize()
 
13
  logger = logging.getLogger(__name__)
14
 
15
  # --- إعداد النموذج ---
16
+ MODEL_NAME = "Salesforce/grappa_large_jnt"
17
+ # يمكنك تغييره لنموذج آخر إذا رغبت
18
 
19
  tokenizer = None
20
  model = None
 
25
  logger.info(f"تحميل النموذج على الجهاز: {device}")
26
 
27
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
28
+ model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
29
  logger.info("تم تحميل النموذج بنجاح")
30
 
31
  initialize()