Alan Liu commited on
Commit
5ae9f66
1 Parent(s): 0829fea
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -44,7 +44,8 @@ def load_model_config(model_id):
44
  model_config['model'] = get_model(model_id, None, None)
45
  module_tensors = get_module_tensors(model_config['model'])
46
  model_config['module_classes'] = classify_module(module_tensors)
47
- except:
 
48
  model_config['model'] = None
49
  model_config['module_classes'] = None
50
 
 
44
  model_config['model'] = get_model(model_id, None, None)
45
  module_tensors = get_module_tensors(model_config['model'])
46
  model_config['module_classes'] = classify_module(module_tensors)
47
+ except Exception as e:
48
+ st.warning(e)
49
  model_config['model'] = None
50
  model_config['module_classes'] = None
51
 
requirements.txt CHANGED
@@ -2,5 +2,5 @@ accelerate @ git+https://github.com/huggingface/accelerate
2
  huggingface_hub
3
  numpy
4
  pandas
5
- requests==2.31.0
6
  streamlit
 
2
  huggingface_hub
3
  numpy
4
  pandas
5
+ requests
6
  streamlit