Update app.py

app.py

@@ -27,8 +27,8 @@ def get_response(prompt:str):
 def create_handler(name: str):
     def handler():
         try:
-            prompt = request.json.get('prompt', None)
             if name == 'write':
+                prompt = request.json['prompt']
                 response = get_response(prompt)
                 return response