Update app.py

app.py

@@ -44,6 +44,7 @@ def wandb_log(prompt, completion):
 def invoke(prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+    completion = ""
     try:
         #completion = generation_model.predict(prompt = prompt,
         #                                      max_output_tokens = config["max_output_tokens"],
@@ -62,7 +63,6 @@ def invoke(prompt):
         if (completion.text != None):
             completion = completion.text
     except Exception as e:
-        completion = e
         raise gr.Error(e)
     finally:
         wandb_log(prompt, completion)