jbilcke-hf HF staff commited on
Commit
8f2d0e0
·
1 Parent(s): 8ca220b

Update base/app.py

Browse files
Files changed (1) hide show
  1. base/app.py +7 -1
base/app.py CHANGED
@@ -33,7 +33,13 @@ unet.eval()
33
  vae.eval()
34
  text_encoder_one.eval()
35
 
36
- def infer(secret_token, prompt, seed_inp, ddim_steps,cfg, infer_type):
 
 
 
 
 
 
37
  if secret_token != SECRET_TOKEN:
38
  raise gr.Error(f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
39
 
 
33
  vae.eval()
34
  text_encoder_one.eval()
35
 
36
+ #def infer(secret_token, prompt, seed_inp, ddim_steps,cfg, infer_type):
37
+ def infer(secret_token, prompt);
38
+ seed_inp = -1
39
+ ddim_steps = 50
40
+ cfg = 7.5
41
+ infer_type = "ddim"
42
+
43
  if secret_token != SECRET_TOKEN:
44
  raise gr.Error(f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
45