AlekseyKorshuk commited on
Commit
a751c70
1 Parent(s): 4cc4bad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -219,7 +219,8 @@ if st.button("Run"):
219
  try:
220
  tokenizer = AutoTokenizer.from_pretrained(f"huggingartists/{model_name}")
221
  model = AutoModelForCausalLM.from_pretrained(f"huggingartists/{model_name}")
222
- except:
 
223
  st.markdown(f"Model for this artist does not exist yet. Create it in just 5 min with [Colab Notebook](https://colab.research.google.com/github/AlekseyKorshuk/huggingartists/blob/master/huggingartists-demo.ipynb):")
224
  st.markdown(
225
  """
219
  try:
220
  tokenizer = AutoTokenizer.from_pretrained(f"huggingartists/{model_name}")
221
  model = AutoModelForCausalLM.from_pretrained(f"huggingartists/{model_name}")
222
+ except Exception as ex:
223
+ st.markdown(ex)
224
  st.markdown(f"Model for this artist does not exist yet. Create it in just 5 min with [Colab Notebook](https://colab.research.google.com/github/AlekseyKorshuk/huggingartists/blob/master/huggingartists-demo.ipynb):")
225
  st.markdown(
226
  """