Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def infer(prompt, negative, scale, model_id=MODELS[0], hf_token="", images=[]):
|
|
33 |
if not hf_token: hf_token = os.getenv("HF_TOKEN", None)
|
34 |
client = InferenceClient(model=model_id, token=hf_token)
|
35 |
for i in range(4):
|
36 |
-
image = client.text_to_image(prompt=prompt, negative_prompt=negative, guidance_scale=scale)
|
37 |
if image: images.append(image)
|
38 |
yield images
|
39 |
#payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
|
|
|
33 |
if not hf_token: hf_token = os.getenv("HF_TOKEN", None)
|
34 |
client = InferenceClient(model=model_id, token=hf_token)
|
35 |
for i in range(4):
|
36 |
+
image = client.text_to_image(prompt=prompt, negative_prompt=negative, guidance_scale=scale, seed=gen_random_seed())
|
37 |
if image: images.append(image)
|
38 |
yield images
|
39 |
#payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
|