Update app.py

app.py

@@ -40,12 +40,12 @@ 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_body = json.loads(response.get("body").read())
-    #completion = response_body["completion"]
-    #wandb_log(prompt, completion)
-    #return completion
-    return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/amazon-bedrock-llm/tree/main."
+    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"]
+    wandb_log(prompt, completion)
+    return completion
+    #return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/amazon-bedrock-llm/tree/main."
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using <a href='https://aws.amazon.com/bedrock/'>Amazon Bedrock</a> API 
                  with <a href='https://www.anthropic.com/'>Anthropic</a> Claude 2 foundation model.