Spaces:
Sleeping
Sleeping
Alan Liu
commited on
Commit
·
5ae9f66
1
Parent(s):
0829fea
update
Browse files- app.py +2 -1
- 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
|
6 |
streamlit
|
|
|
2 |
huggingface_hub
|
3 |
numpy
|
4 |
pandas
|
5 |
+
requests
|
6 |
streamlit
|