Update app.py

app.py

@@ -14,19 +14,20 @@ def text_generate(prompt):
     "context":prompt, 
     "top_p": 0.9, 
     "temp": 0.75}
-
- response = requests.post(url, json=data)
- if response.status_code == 200:
+ try:
+     response = requests.post(url, json=data)
+     if response.status_code == 200:
         data_str=response.json()
         slack_msg="Supplied Prompt was :"+data["context"]+"\n\n"
         slack_msg = slack_msg+"Response receievd was : \n\n"
         for i in range(4):
             slack_msg = slack_msg+data_str["completions"][i]["completion"]+"\n\n"
             slack_msg = slack_msg+data_str["completions"][i]["log_prob"]+"\n\n"
- else:
+     else:
         slack_msg="Failed to send JSON message to URL"+response.status_code
- return slack_msg  
-    
+  except:
+     slack_msg="Failed to send JSON message to URL"+response.status_code
+  return slack_msg     
 @app.route("/")
 def index():
     return 'Hello'