John6666 commited on
Commit
a6002ee
·
verified ·
1 Parent(s): d36e3ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}