Update app.py

app.py

@@ -58,8 +58,10 @@ def invoke(prompt):
                                                                    "top_p": config["top_p"],
                                                                }).text
     except Exception as e:
+        completion = e
         raise gr.Error(e)
-    wandb_log(prompt, completion)
+    finally:
+        wandb_log(prompt, completion)
     return completion
     #return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."