dralsarrani commited on
Commit
e93a7d4
Β·
verified Β·
1 Parent(s): 85d43b2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -27,7 +27,7 @@ app.add_middleware(
27
 
28
  # ─── LOAD MODEL ───────────────────────────────────────────────────────────────
29
 
30
- HF_MODEL = os.getenv("HF_MODEL", "dralsarrani/arabert-sentiment")
31
  DEVICE = 0 if torch.cuda.is_available() else -1
32
 
33
  print(f"Loading model: {HF_MODEL}")
 
27
 
28
  # ─── LOAD MODEL ───────────────────────────────────────────────────────────────
29
 
30
+ HF_MODEL = os.getenv("HF_MODEL", "dralsarrani/arareview")
31
  DEVICE = 0 if torch.cuda.is_available() else -1
32
 
33
  print(f"Loading model: {HF_MODEL}")