Update app.py

app.py

@@ -46,11 +46,17 @@ def invoke(prompt):
         modelId = config["model"]
         accept = "application/json"
         contentType = "application/json"
-        response = bedrock_runtime.invoke_model(body = body, modelId = modelId, accept = accept, contentType = contentType)
+        
+        response = bedrock_runtime.invoke_model(body = body, 
+                                                modelId = modelId, 
+                                                accept = accept, 
+                                                contentType = contentType)
+        
         response_body = json.loads(response.get("body").read())
         completion = response_body["completion"]
     except Exception as e:
         completion = e
+        
         raise gr.Error(e)
     finally:
         wandb_log(prompt, completion)