Update app.py

app.py

@@ -42,18 +42,21 @@ def wandb_log(prompt, completion):
     wandb.finish()
 
 def invoke(prompt):
-    #completion = generation_model.predict(prompt = prompt,
-    #                                      max_output_tokens = config["max_output_tokens"],
-    #                                      temperature = config["temperature"],
-    #                                      top_k = config["top_k"],
-    #                                      top_p = config["top_p"],
-    #                                     ).text
-    completion = generation_model.generate_content(prompt, generation_config = {
-                                                               "max_output_tokens": config["max_output_tokens"],
-                                                               "temperature": config["temperature"],
-                                                               "top_k": config["top_k"],
-                                                               "top_p": config["top_p"],
-                                                           }).text
+    try:
+        #completion = generation_model.predict(prompt = prompt,
+        #                                      max_output_tokens = config["max_output_tokens"],
+        #                                      temperature = config["temperature"],
+        #                                      top_k = config["top_k"],
+        #                                      top_p = config["top_p"],
+        #                                     ).text
+        completion = generation_model.generate_content(prompt, generation_config = {
+                                                                   "max_output_tokens": config["max_output_tokens"],
+                                                                   "temperature": config["temperature"],
+                                                                   "top_k": config["top_k"],
+                                                                   "top_p": config["top_p"],
+                                                               }).text
+    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/google-vertex-ai-llm/tree/main."