Update app.py

app.py

@@ -3,11 +3,9 @@ import requests
 app = Flask(__name__)
 @app.route('/claude', methods=['POST'])
 def claude():
-    model = request.get_json().get("model", "claude-2")
-
-    API_KEY = request.get_json().get("api_key")
-
-    messages = request.get_json().get("messages")
+    messages = request.json['messages']
+    API_KEY = request.json['API_KEY']
+    model = 'claude-2'
     headers = {'Authorization': API_KEY}
     prompt = ""
     for i in messages: