Update app.py

app.py

@@ -30,19 +30,24 @@ def wandb_log(prompt, completion):
     wandb.finish()
 
 def invoke(prompt):
-    body = json.dumps({"prompt": "\n\nHuman: " + prompt + "\n\nAssistant: ",
-                       "max_tokens_to_sample": config["max_tokens_to_sample"],
-                       "temperature": config["temperature"],
-                       "top_k": config["top_k"],
-                       "top_p": config["top_p"],
-                       "stop_sequences": ["\n\nHuman: "]
-                      })
-    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"]
+    if (prompt == ""):
+        raise gr.Error("Prompt is required.")
+    try:
+        body = json.dumps({"prompt": "\n\nHuman: " + prompt + "\n\nAssistant: ",
+                           "max_tokens_to_sample": config["max_tokens_to_sample"],
+                           "temperature": config["temperature"],
+                           "top_k": config["top_k"],
+                           "top_p": config["top_p"],
+                           "stop_sequences": ["\n\nHuman: "]
+                          })
+        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"]
+    except Exception as e:
+        raise gr.Error(e)
     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."