Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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()))
|