Update model.py

model.py

@@ -55,20 +55,23 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
 #     texts.append(f'{message} [/INST]')
 #     return ''.join(texts)
 
-
 def get_prompt(message: str, chat_history: list[tuple[str, str]], system_prompt: str) -> str:
     texts = [f'{system_prompt}\n']
 
-    for user_input, response in chat_history[:-1]:
-        texts.append(f'{user_input} {response}\n')
+    if chat_history:
+        for user_input, response in chat_history[:-1]:
+            texts.append(f'{user_input} {response}\n')
 
-    # Getting the user input and response from the last tuple in the chat history
-    last_user_input, last_response = chat_history[-1]
-    texts.append(f' input: {last_user_input} {last_response} {message} Response: ')
+        # Getting the user input and response from the last tuple in the chat history
+        last_user_input, last_response = chat_history[-1]
+        texts.append(f' input: {last_user_input} {last_response} {message} Response: ')
+    else:
+        texts.append(f' input: {message} Response: ')
 
     return ''.join(texts)
 
 
+
 def get_input_token_length(message: str, chat_history: list[tuple[str, str]], system_prompt: str) -> int:
     prompt = get_prompt(message, chat_history, system_prompt)
     input_ids = tokenizer([prompt], return_tensors='np', add_special_tokens=False)['input_ids']