Update app.py

app.py

@@ -26,7 +26,7 @@ max_new_tokens=1000
 top_p=0.92
 repetition_penalty=1.7
 
-model_name = "Inagua/code-model"
+model_name = "Inagua/code-model-2"
 
 llm = LLM(model_name, max_model_len=4096)
 
@@ -134,7 +134,8 @@ class MistralChatBot:
         self.system_prompt = system_prompt
 
     def predict(self, user_message, context):
-        detailed_prompt = """### Question ###\n""" + user_message + "\n\n### Contexte ###\n" + context + "\n\n### Formule ###\n"
+        #detailed_prompt = """### Question ###\n""" + user_message + "\n\n### Contexte ###\n" + context + "\n\n### Formule ###\n"
+        detailed_prompt = """### Question ###\n""" + user_message + "\n\n### Field ###\n"
         prompts = [detailed_prompt]
         outputs = llm.generate(prompts, sampling_params, use_tqdm = False)
         generated_text = outputs[0].outputs[0].text