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"
|