jbilcke-hf HF staff commited on
Commit
2d87969
·
1 Parent(s): 92ff4bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -83,7 +83,9 @@ def generate(prompt: str,
83
  num_inference_steps_refiner: int = 50,
84
  secret_token: str = '') -> PIL.Image.Image:
85
  if secret_token != SECRET_TOKEN:
86
- raise ValueError("Invalid secret token!")
 
 
87
  generator = torch.Generator().manual_seed(seed)
88
 
89
  if not use_negative_prompt:
 
83
  num_inference_steps_refiner: int = 50,
84
  secret_token: str = '') -> PIL.Image.Image:
85
  if secret_token != SECRET_TOKEN:
86
+ raise gr.Error(
87
+ f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
88
+
89
  generator = torch.Generator().manual_seed(seed)
90
 
91
  if not use_negative_prompt: