rlancemartin commited on
Commit
65183f6
1 Parent(s): 118cf75

Error with hosted HF model

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -337,8 +337,10 @@ st.sidebar.image("img/diagnostic.jpg")
337
 
338
  oai_api_key = st.sidebar.text_input("`OpenAI API Key:`", type="password")
339
  ant_api_key = st.sidebar.text_input("`(Optional) Anthropic API Key:`", type="password")
 
340
  os.environ["OPENAI_API_KEY"] = oai_api_key
341
  os.environ["ANTHROPIC_API_KEY"] = ant_api_key
 
342
 
343
  with st.sidebar.form("user_input"):
344
 
@@ -359,8 +361,9 @@ with st.sidebar.form("user_input"):
359
  model = st.radio("`Choose model`",
360
  ("gpt-3.5-turbo",
361
  "gpt-4",
362
- "anthropic",
363
- "flan-t5-xl"),
 
364
  index=0)
365
 
366
  retriever_type = st.radio("`Choose retriever`",
 
337
 
338
  oai_api_key = st.sidebar.text_input("`OpenAI API Key:`", type="password")
339
  ant_api_key = st.sidebar.text_input("`(Optional) Anthropic API Key:`", type="password")
340
+ hf_api_key = st.sidebar.text_input("`(Optional) HuggingFace API Token:`", type="password")
341
  os.environ["OPENAI_API_KEY"] = oai_api_key
342
  os.environ["ANTHROPIC_API_KEY"] = ant_api_key
343
+ os.environ["HUGGINGFACEHUB_API_TOKEN"] = hf_api_key
344
 
345
  with st.sidebar.form("user_input"):
346
 
 
361
  model = st.radio("`Choose model`",
362
  ("gpt-3.5-turbo",
363
  "gpt-4",
364
+ "anthropic"),
365
+ # Error raised by inference API: Model google/flan-t5-xl time out
366
+ #"flan-t5-xl"),
367
  index=0)
368
 
369
  retriever_type = st.radio("`Choose retriever`",