浏览代码

add upper and lower limit to countables

kevin 1 年之前
父节点
当前提交
f753b946cb
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -130,9 +130,9 @@ def gpt():
         has_named_params = 'payload' in json_payload
         if 'payload' in json_payload:
             if 'predict_q' in json_payload:
-                predict_q = int(json_payload['predict_q'])
+                predict_q = 5 if json_payload['predict_q'] > 5 else 0 if json_payload['predict_q'] < 0 else json_payload['predict_q']
             if 'num_choices' in json_payload:
-                num_choices = 5 if json_payload['num_choices'] > 5 else json_payload['num_choices']
+                num_choices = 5 if json_payload['num_choices'] > 5 else 1 if json_payload['num_choices'] < 1 else json_payload['num_choices']
             if 'use_video' in json_payload:
                 use_video = json_payload['use_video'] == "1"
             if 'chat_model' in json_payload: