소스 검색

all user input lowercase

kevin 1 년 전
부모
커밋
54a4625834
2개의 변경된 파일8개의 추가작업 그리고 4개의 파일을 삭제
  1. 1 1
      convert.py
  2. 7 3
      main.py

+ 1 - 1
convert.py

@@ -16,7 +16,7 @@ def alpaca_to_chatgpt(jsf):
             "messages":
                 [
                     {"role": "system", "content": SYSTEM_MESSAGE},
-                    {"role": "user", "content": j['instruction']},
+                    {"role": "user", "content": j['instruction'].lower()},
                     {"role": "assistant", "content": j['output']}
                 ]
         }

+ 7 - 3
main.py

@@ -133,6 +133,7 @@ def gpt():
                 use_video = json_payload['use_video'] == "1"
             if 'chat_model' in json_payload:
                 chat_model = json_payload['chat_model']
+                max_resp_token = 2048
             if 'translate' in json_payload:
                 chat_messages = app.translate_messages.copy()
                 json_payload['payload'][-1]['content'] = json_payload['payload'][-1][
@@ -198,9 +199,12 @@ def gpt():
         else:
             chat_messages = app.chat_messages.copy()
             json_payload = [json_payload]
-    json_payload = json_payload[-7:]
+    json_payload = json_payload[-5:]
     for message in json_payload:
-        content = message['content']
+        if message['role'] == 'user':
+            content = message['content'].lower()
+        else:
+            content = message['content']
         content_arr = content.split(" ")
         new_content_arr = content[:max_char_msg].split(" ")
         new_content_len = len(new_content_arr)
@@ -230,7 +234,7 @@ def gpt():
                 result = {"url": url_for('download_file', name="test.mp4", _external=True),
                           "message": {"role": "assistant", "content": json_response.choices[0].message.content}}
             else:
-                result = {"url": "", "message": {"role": "assistant", "content": json_response.choices[0].message.content}}
+                result = {"role": "assistant", "content": json_response.choices[0].message.content}
         else:
             result = {"role": "assistant", "content": json_response.choices[0].message.content}
     except Exception as error_print: