Upload app.py

app.py

@@ -2,7 +2,7 @@
 # pylint: disable=broad-exception-caught, redefined-outer-name, missing-function-docstring, missing-module-docstring, too-many-arguments, line-too-long, invalid-name, redefined-builtin, redefined-argument-from-local
 # import gradio as gr
 
-# model_name = "models/THUDM/chatglm2-6b-int4"
+# model_name = "fb700/chatglm-fitness-RLHF"
 # gr.load(model_name).lauch()
 
 # %%writefile demo-4bit.py
@@ -26,7 +26,7 @@ except Exception:
     logger.warning("Windows, cant run time.tzset()")
 
 # model_name = "THUDM/chatglm2-6b"  # 7x?G
-model_name = "THUDM/chatglm2-6b-int4"  # 3.9G
+model_name = "fb700/chatglm-fitness-RLHF"  # 3.9G
 
 RETRY_FLAG = False
 
@@ -134,7 +134,7 @@ def predict(
         yield chatbot, history, past_key_values
 
 
-def trans_api(input, max_length=4096, top_p=0.8, temperature=0.2):
+def trans_api(input, max_length=40960, top_p=0.8, temperature=0.2):
     if max_length < 10:
         max_length = 4096
     if top_p < 0.1 or top_p > 1: