Update endpoints.py

endpoints.py

@@ -32,11 +32,12 @@ def LLM(llm_name, length):
 # base_model = AutoModelForCausalLM.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
 # Mistral 7B
 # mistral_llm = LLM("mistralai/Mistral-7B-v0.1",30000)
-mistral_llm = LLM("microsoft/phi-2",2000)
+# mistral_llm = LLM("microsoft/phi-2",2000)
 
 # WizardCoder 13B
 # wizard_llm = LLM("WizardLM/WizardCoder-Python-13B-V1.0",8000)
 wizard_llm =  LLM("WizardLM/WizardCoder-3B-V1.0",4000)
+mistral_llm = wizard_llm
 # hf_llm = HuggingFacePipeline(pipeline=pipe)
 
 def ask_model(model, prompt):