Update app.py

app.py

@@ -133,6 +133,7 @@ def hermes_generate_response(msg_prompt: str) -> dict:
 
     
 def mistral_generate_response(msg_prompt, persona_desc_prompt):
+    
     user_prompt = f'{msg_prompt} [/INST]'
     persona_prompt = f'{persona_desc_prompt} [/INST]'
     prompt_template = f'''[INST] Instruction:{persona_prompt} [INST] {user_prompt}'''
@@ -144,6 +145,7 @@ def mistral_generate_response(msg_prompt, persona_desc_prompt):
     generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
     decoded = tokenizer.batch_decode(generated_ids)
     response = (decoded[0])
+    
     return response