Spaces:

bstraehle
/

vertex-ai

Sleeping

bstraehle commited on Dec 18, 2023

Commit

f45a806

1 Parent(s): 22a1dd5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,6 +42,8 @@ def wandb_log(prompt, completion):
     wandb.finish()
 def invoke(prompt):
     try:
         #completion = generation_model.predict(prompt = prompt,
         #                                      max_output_tokens = config["max_output_tokens"],

     wandb.finish()
 def invoke(prompt):
+    if (prompt == ""):
+        raise gr.Error("Prompt is required.")
     try:
         #completion = generation_model.predict(prompt = prompt,
         #                                      max_output_tokens = config["max_output_tokens"],