Update app.py

app.py

@@ -39,7 +39,10 @@ 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, timeout=(20, 100)).json()
+    try:
+        response = requests.request("POST", url, headers=headers, data=payload, timeout=(20, 100)).json()
+    except Exception as e:
+        print('Timeout! Please wait a few minutes and retry')
     
     if response['status'] == 1:
         return 'Please give smaller text than max_tokens or give larger max_tokens.'