Update main.py

main.py

@@ -53,6 +53,8 @@ def generate_text(request: Request, input_data: InputData):
     top_p = input_data.top_p
     repetition_penalty = input_data.repetition_penalty
     input_text = generate_input_text(input_data)
+    if input_data.systemStyle != '': 
+        input_text = input_text + '(' + input_data.systemStyle + ')'    
     max_new_tokens = min(max_new_tokens, 29500 - len(input_text))
     print(f"{datetime.now()} - Input Text: {input_text}")
     history = []