Update app.py

app.py

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