Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,8 @@ from transformers import AutoConfig # AutoConfig is from the Hugging Face Trans
|
|
| 4 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
|
| 5 |
|
| 6 |
# A list of model names to start with. These are names of popular models from the Hugging Face library.
|
| 7 |
-
model_list = ["bert-base-uncased", "TigerResearch/tigerbot-70b-chat-4bit-v2", "openai/whisper-large-v3", "google/switch-c-2048", "gpt2","mosaicml/mpt-30b",
|
|
|
|
| 8 |
|
| 9 |
# Function to add a new model to the list.
|
| 10 |
def add_model_to_list(new_model):
|
|
|
|
| 4 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
|
| 5 |
|
| 6 |
# A list of model names to start with. These are names of popular models from the Hugging Face library.
|
| 7 |
+
model_list = ["bert-base-uncased", "TigerResearch/tigerbot-70b-chat-4bit-v2", "openai/whisper-large-v3", "openai/shap-e","google/switch-c-2048", "gpt2","mosaicml/mpt-30b",
|
| 8 |
+
"google/flan-t5-small","mosaicml/mpt-7b","distilbert-base-uncased,","microsoft/speecht5_tts","suno/bark-small","suno/bark",]
|
| 9 |
|
| 10 |
# Function to add a new model to the list.
|
| 11 |
def add_model_to_list(new_model):
|