Update server.py

server.py

@@ -70,12 +70,12 @@ args = parser.parse_args()
 summarization_model = (
     args.summarization_model
     if args.summarization_model
-    else DEFAULT_SUMMARIZATION_MODEL
+    else "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
 )
 classification_model = (
     args.classification_model
     if args.classification_model
-    else DEFAULT_CLASSIFICATION_MODEL
+    else "nateraw/bert-base-uncased-emotion"
 )
 
 device_string = "cpu"
@@ -300,7 +300,32 @@ def api_summarize():
     if "text" not in data or not isinstance(data["text"], str):
         abort(400, '"text" is required')
 
-    params = DEFAULT_SUMMARIZE_PARAMS.copy()
+    params = {
+    "temperature": 1.0,
+    "repetition_penalty": 1.0,
+    "max_length": 500,
+    "min_length": 200,
+    "length_penalty": 1.5,
+    "bad_words": [
+        "\n",
+        '"',
+        "*",
+        "[",
+        "]",
+        "{",
+        "}",
+        ":",
+        "(",
+        ")",
+        "<",
+        ">",
+        "Â",
+        "The text ends",
+        "The story ends",
+        "The text is",
+        "The story is",
+    ],
+}
 
     if "params" in data and isinstance(data["params"], dict):
         params.update(data["params"])