Update app.py

app.py

@@ -29,8 +29,11 @@ def create_handler(name: str):
         try:
             if name == 'write':
                 prompt = request.json['prompt']
-                response = get_response(prompt)
-                return response
+                try:
+                    response = get_response(prompt)
+                    return response
+                except Exception as e:
+                    return str(e)
 
         except Exception as error:
             return jsonify({"error": str(error)}), 500