melvinalves commited on
Commit
0ffe325
·
verified ·
1 Parent(s): 596916b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import os
2
  import numpy as np
3
  import torch, joblib, streamlit as st
4
  from transformers import AutoTokenizer, AutoModel
@@ -31,6 +31,8 @@ def load_hf_model(model_name: str):
31
  @st.cache_resource
32
  def load_local_model(fname: str):
33
  full_path = hf_cached(f"models/{fname}")
 
 
34
  return load_model(full_path, compile=False)
35
 
36
  # ---------- Carregamento dos modelos locais (.keras) ----------
 
1
+ import os, pathlib
2
  import numpy as np
3
  import torch, joblib, streamlit as st
4
  from transformers import AutoTokenizer, AutoModel
 
31
  @st.cache_resource
32
  def load_local_model(fname: str):
33
  full_path = hf_cached(f"models/{fname}")
34
+ print(">>", pathlib.Path(full_path).name,
35
+ "tamanho:", os.path.getsize(full_path), "bytes")
36
  return load_model(full_path, compile=False)
37
 
38
  # ---------- Carregamento dos modelos locais (.keras) ----------