Update app.py

app.py

@@ -27,12 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained("google/muril-base-cased")
 model = AutoModelForMaskedLM.from_pretrained("google/muril-base-cased")
 
 # default_persist_directory = './chroma_HF/'
-list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
-    "google/gemma-7b-it","google/gemma-2b-it", \
-    "HuggingFaceH4/zephyr-7b-beta", "HuggingFaceH4/zephyr-7b-gemma-v0.1", \
-    "meta-llama/Llama-2-7b-chat-hf", "microsoft/phi-2", \
-    "TinyLlama/TinyLlama-1.1B-Chat-v1.0", "mosaicml/mpt-7b-instruct", "tiiuae/falcon-7b-instruct", \
-    "google/flan-t5-xxl"
+list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "DeepSeek-R1-Distill-Qwen-1.5B","mistralai/Mistral-7B-Instruct-v0.1"
 ]
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]