Rafal commited on
Commit
c6292bf
·
1 Parent(s): 2fb6641

Removed LLAMA7b and FALCON for now, they are too slow to load

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -871,8 +871,8 @@ with gr.Blocks(theme=soft, title="Social Bias Testing in Language Models",
871
  with gr.Column():
872
  gen_title = gr.Markdown("### Select Tested Model", visible=True)
873
 
874
- # Tested Model Selection - "openlm-research/open_llama_7b"
875
- tested_model_name = gr.Dropdown( ["bert-base-uncased","bert-large-uncased","gpt2","gpt2-medium","gpt2-large","emilyalsentzer/Bio_ClinicalBERT","microsoft/biogpt","openlm-research/open_llama_3b", "openlm-research/open_llama_7b", "tiiuae/falcon-7b"], value="bert-base-uncased",
876
  multiselect=None,
877
  interactive=True,
878
  label="Tested Language Model",
 
871
  with gr.Column():
872
  gen_title = gr.Markdown("### Select Tested Model", visible=True)
873
 
874
+ # Tested Model Selection - "openlm-research/open_llama_7b", "tiiuae/falcon-7b"
875
+ tested_model_name = gr.Dropdown( ["bert-base-uncased","bert-large-uncased","gpt2","gpt2-medium","gpt2-large","emilyalsentzer/Bio_ClinicalBERT","microsoft/biogpt","openlm-research/open_llama_3b"], value="bert-base-uncased",
876
  multiselect=None,
877
  interactive=True,
878
  label="Tested Language Model",