Mazenbs commited on
Commit
8512af3
ยท
verified ยท
1 Parent(s): 6c90697

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -40,11 +40,12 @@ device = torch.device("cpu")
40
  def load_model():
41
  global model, config
42
  if model is None or config is None:
43
- print("๐Ÿš€ ุชุญู…ูŠู„ ุงู„ู†ู…ูˆุฐุฌ ู„ุฃูˆู„ ู…ุฑุฉ...")
44
  model_path = snapshot_download(
45
  repo_id="shahad-alh/arabichar-finetuned-v2",
46
  local_dir="./model"
47
  )
 
48
  config = AutoConfig.from_pretrained(model_path, trust_remote_code=True)
49
  model = AutoModelForImageClassification.from_pretrained(
50
  model_path,
 
40
  def load_model():
41
  global model, config
42
  if model is None or config is None:
43
+ print("๐Ÿš€ ุชุญู…ูŠู„ ุงู„ู†ู…ูˆุฐุฌ ู„ุฃูˆู„ ู…ุฑุฉ ู…ู† Hugging Face Hub...")
44
  model_path = snapshot_download(
45
  repo_id="shahad-alh/arabichar-finetuned-v2",
46
  local_dir="./model"
47
  )
48
+ # ุชุญู…ูŠู„ ุงู„ูƒูˆุฏ ุงู„ู…ุฎุตุต ู…ุน trust_remote_code=True
49
  config = AutoConfig.from_pretrained(model_path, trust_remote_code=True)
50
  model = AutoModelForImageClassification.from_pretrained(
51
  model_path,