Update app.py

app.py

@@ -12,6 +12,9 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
     temperature, topk, topp):
     global APIKEY
     global APISECRET
+    
+    if text == '':
+        return 'Input should not be empty!'
 
     url = 'https://wudao.aminer.cn/os/api/api/v2/completions_130B'
 
@@ -36,14 +39,12 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
         'Content-Type': 'application/json'
     }
 
-    response = requests.request("POST", url, headers=headers, data=payload)
-    
-    print(response.json())
+    response = requests.request("POST", url, headers=headers, data=payload).json()
     
-    if response.json()['status'] == 1:
+    if response['status'] == 1:
         return 'Please give smaller text than max_tokens or give larger max_tokens.'
     
-    answer = response.json()['result']['output']['raw']
+    answer = response['result']['output']['raw']
     if isinstance(answer, list):
         answer = answer[0]