miwojc commited on
Commit
d4fb97b
1 Parent(s): 3cf2155

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ st.markdown(
73
  The models were trained with Jax and Flax using TPUs as part of the [Flax/Jax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104) organised by HuggingFace.
74
  """
75
  )
76
- model_name = MODEL_NAME
77
  ALL_PROMPTS = list(PROMPT_LIST.keys()) + ["Custom"]
78
  prompt = st.selectbox("Prompt", ALL_PROMPTS, index=len(ALL_PROMPTS) - 1)
79
  if prompt == "Custom":
73
  The models were trained with Jax and Flax using TPUs as part of the [Flax/Jax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104) organised by HuggingFace.
74
  """
75
  )
76
+ model_name = MODEL_URL
77
  ALL_PROMPTS = list(PROMPT_LIST.keys()) + ["Custom"]
78
  prompt = st.selectbox("Prompt", ALL_PROMPTS, index=len(ALL_PROMPTS) - 1)
79
  if prompt == "Custom":