Update endpoints.py

endpoints.py

@@ -36,14 +36,13 @@ def LLM(llm_name, length):
 
 # 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
+wizard_llm =  LLM("WizardLM/WizardCoder-1B-V1.0",4000)
 # hf_llm = HuggingFacePipeline(pipeline=pipe)
 
 def ask_model(model, prompt):
-    if(model == 'mistral'):
-        result = mistral_llm(prompt)
-        return result
+    # if(model == 'mistral'):
+    #     result = mistral_llm(prompt)
+    #     return result
     if(model == 'wizard'):
          result =  wizard_llm(prompt)
          return result