patrickvonplaten commited on
Commit
47fe7d6
·
1 Parent(s): de313da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def inference(
39
  pipe = DiffusionPipeline.from_pretrained(repo_id, revision=revision, torch_dtype=dtype)
40
  pipe.to(torch_device)
41
 
42
- return pipe(prompt, generator=generator, num_inference_steps=25).images
43
  except Exception as e:
44
  url = f"https://huggingface.co/{repo_id}/discussions/{discuss_nr}"
45
  message = f"There is a problem with your diffusers weights of the PR: {url}. Error message: \n"
 
39
  pipe = DiffusionPipeline.from_pretrained(repo_id, revision=revision, torch_dtype=dtype)
40
  pipe.to(torch_device)
41
 
42
+ return pipe(prompt, generator=generator, num_inference_steps=25).images, f"Done. Seed: {seed}"
43
  except Exception as e:
44
  url = f"https://huggingface.co/{repo_id}/discussions/{discuss_nr}"
45
  message = f"There is a problem with your diffusers weights of the PR: {url}. Error message: \n"