updated default model
Browse files
app.py
CHANGED
|
@@ -7,7 +7,14 @@ from utils import clear_uploader, undo, restart
|
|
| 7 |
|
| 8 |
|
| 9 |
share_keys = ["messages", "model_name"]
|
| 10 |
-
MODELS = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
st.set_page_config(
|
| 13 |
page_title="LLM",
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
share_keys = ["messages", "model_name"]
|
| 10 |
+
MODELS = [
|
| 11 |
+
"google/flan-t5-small",
|
| 12 |
+
"google/flan-t5-base",
|
| 13 |
+
"google/flan-t5-large",
|
| 14 |
+
"google/flan-t5-xl",
|
| 15 |
+
"google/flan-t5-xxl",
|
| 16 |
+
]
|
| 17 |
+
default_model = "google/flan-t5-small"
|
| 18 |
|
| 19 |
st.set_page_config(
|
| 20 |
page_title="LLM",
|