Update app.py
Browse files
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(
|
12 |
-
prompt = prompt2
|
13 |
seed: int = 0
|
14 |
guidance_scale: float = 15.0
|
15 |
num_inference_steps: int = 64
|
16 |
|
17 |
-
|
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,
|