FaiziRBLX commited on
Commit
afce7ee
Β·
verified Β·
1 Parent(s): 2cfa45e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -20,10 +20,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
20
  logger.info(f"Device: {device}")
21
 
22
  # ── Cek model file ────────────────────────────────────────
23
- logger.info(f"model.pt ada: {os.path.exists('model.pt')}")
24
- if not os.path.exists('model.pt'):
25
  raise FileNotFoundError("model.pt tidak ditemukan! Upload dulu ke Space.")
26
- logger.info(f"model.pt size: {os.path.getsize('model.pt') / 1e6:.1f} MB")
27
 
28
  # ── Load tokenizer ────────────────────────────────────────
29
  logger.info("Loading tokenizer...")
@@ -33,7 +33,7 @@ logger.info("Tokenizer OK")
33
 
34
  # ── Load model ────────────────────────────────────────────
35
  logger.info("Loading checkpoint...")
36
- checkpoint = torch.load("model.pt", map_location='cpu', weights_only=False)
37
  logger.info(f"Checkpoint keys: {list(checkpoint.keys())}")
38
 
39
  logger.info("Building model...")
 
20
  logger.info(f"Device: {device}")
21
 
22
  # ── Cek model file ────────────────────────────────────────
23
+ logger.info(f"model.pt ada: {os.path.exists('indonesian_llm_model (39).pt')}")
24
+ if not os.path.exists('indonesian_llm_model (39).pt'):
25
  raise FileNotFoundError("model.pt tidak ditemukan! Upload dulu ke Space.")
26
+ logger.info(f"model.pt size: {os.path.getsize('indonesian_llm_model (39).pt') / 1e6:.1f} MB")
27
 
28
  # ── Load tokenizer ────────────────────────────────────────
29
  logger.info("Loading tokenizer...")
 
33
 
34
  # ── Load model ────────────────────────────────────────────
35
  logger.info("Loading checkpoint...")
36
+ checkpoint = torch.load("indonesian_llm_model (39).pt", map_location='cpu', weights_only=False)
37
  logger.info(f"Checkpoint keys: {list(checkpoint.keys())}")
38
 
39
  logger.info("Building model...")