Update app.py

app.py

@@ -32,6 +32,7 @@ def wandb_log(prompt, completion):
 def invoke(prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+    completion = ""
     try:
         body = json.dumps({"prompt": "\n\nHuman: " + prompt + "\n\nAssistant: ",
                            "max_tokens_to_sample": config["max_tokens_to_sample"],
@@ -47,7 +48,6 @@ def invoke(prompt):
         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)