Illustre commited on
Commit
1f1d405
1 Parent(s): 0133b34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -8,19 +8,12 @@ from model import Model
8
  from settings import CACHE_EXAMPLES, MAX_SEED
9
  from utils import randomize_seed_fn
10
 
11
- def inference(prompt2: str) -> str:
12
- prompt = prompt2
13
  seed: int = 0
14
  guidance_scale: float = 15.0
15
  num_inference_steps: int = 64
16
 
17
- inputs = [
18
- prompt,
19
- seed,
20
- guidance_scale,
21
- num_inference_steps,
22
- ]
23
- return Model.run_text(inputs)
24
 
25
  demo = gr.Interface(
26
  fn=inference,
 
8
  from settings import CACHE_EXAMPLES, MAX_SEED
9
  from utils import randomize_seed_fn
10
 
11
+ def inference(prompt: str) -> str:
 
12
  seed: int = 0
13
  guidance_scale: float = 15.0
14
  num_inference_steps: int = 64
15
 
16
+ return Model.run_text(prompt, seed, guidance_scale, num_inference_steps)
 
 
 
 
 
 
17
 
18
  demo = gr.Interface(
19
  fn=inference,