BigSalmon commited on
Commit
e44cc57
·
1 Parent(s): b8d4d7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import json
2
  import requests
3
- from prompts import PROMPT_LIST
4
  import streamlit as st
5
  import random
6
  headers = {}
@@ -66,8 +65,7 @@ st.markdown(
66
  """Thai GPT-2 demo. Part of the [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/)."""
67
  )
68
  model_name = st.selectbox('Model', (['GPT-2 Base']))
69
- ALL_PROMPTS = list(PROMPT_LIST.keys())+["Custom"]
70
- prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
71
  if prompt == "Custom":
72
  prompt_box = "Enter your text here"
73
  else:
 
1
  import json
2
  import requests
 
3
  import streamlit as st
4
  import random
5
  headers = {}
 
65
  """Thai GPT-2 demo. Part of the [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/)."""
66
  )
67
  model_name = st.selectbox('Model', (['GPT-2 Base']))
68
+ prompt = st.selectbox('Prompt')
 
69
  if prompt == "Custom":
70
  prompt_box = "Enter your text here"
71
  else: