Update app.py

app.py

@@ -27,7 +27,9 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
         "min_gen_length": min_gen_length,
         "sampling_strategy": sampling_strategy,
         "num_beams": num_beams,
-        "max_tokens": out_seq_length
+        "max_tokens": out_seq_length,
+        "no_repeat_ngram": no_repeat_ngram,
+        "seed": seed
     })
 
     headers = {
@@ -36,7 +38,8 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
 
     response = requests.request("POST", url, headers=headers, data=payload)
     
-    print(response.json())
+    if reponse.json()['status'] == 1:
+        return 'Please give smaller text than max_tokens or give larger max_tokens.'
     
     answer = response.json()['result']['output']['raw']
     if isinstance(answer, list):