acecalisto3 commited on
Commit
14d749b
·
verified ·
1 Parent(s): 8edb9cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ st.title("CEEMEESEEK with Model Selection")
44
  model_loaders = {
45
  "Falcon": lambda: load_model("tiiuae/falcon-7b"),
46
  "Flan-T5": lambda: load_model("google/flan-t5-xl"),
47
- # Add more models and their loading functions here
48
  }
49
 
50
  model_option = st.selectbox("Select a Model", list(model_loaders.keys()))
 
44
  model_loaders = {
45
  "Falcon": lambda: load_model("tiiuae/falcon-7b"),
46
  "Flan-T5": lambda: load_model("google/flan-t5-xl"),
47
+ "Flan-T5-Small": lambda: load_model("google/flan-t5-small") # Add a smaller model
48
  }
49
 
50
  model_option = st.selectbox("Select a Model", list(model_loaders.keys()))