Commit
·
47fe7d6
1
Parent(s):
de313da
Update app.py
Browse files
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"
|