Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline
|
|
7 |
import torch
|
8 |
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
-
model_repo_id = "John6666/wai-ani-nsfw-ponyxl-
|
11 |
|
12 |
if torch.cuda.is_available():
|
13 |
torch_dtype = torch.float16
|
@@ -41,7 +41,6 @@ def infer(
|
|
41 |
|
42 |
generator = torch.Generator().manual_seed(seed)
|
43 |
|
44 |
-
# Prepend the specific terms to the prompt and negative prompt
|
45 |
full_prompt = f"{prompt_prefix}, {prompt}"
|
46 |
full_negative_prompt = f"{negative_prompt_prefix}, {negative_prompt}"
|
47 |
|
@@ -67,7 +66,7 @@ css = """
|
|
67 |
with gr.Blocks(css=css) as demo:
|
68 |
with gr.Column(elem_id="col-container"):
|
69 |
gr.Markdown(" # Rainbow Media Anime Generator")
|
70 |
-
gr.Markdown(' ### <a href="https://
|
71 |
|
72 |
result = gr.Image(label="Result", show_label=False)
|
73 |
|
|
|
7 |
import torch
|
8 |
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
+
model_repo_id = "John6666/wai-ani-nsfw-ponyxl-v8-sdxl" # Replace to the model you would like to use
|
11 |
|
12 |
if torch.cuda.is_available():
|
13 |
torch_dtype = torch.float16
|
|
|
41 |
|
42 |
generator = torch.Generator().manual_seed(seed)
|
43 |
|
|
|
44 |
full_prompt = f"{prompt_prefix}, {prompt}"
|
45 |
full_negative_prompt = f"{negative_prompt_prefix}, {negative_prompt}"
|
46 |
|
|
|
66 |
with gr.Blocks(css=css) as demo:
|
67 |
with gr.Column(elem_id="col-container"):
|
68 |
gr.Markdown(" # Rainbow Media Anime Generator")
|
69 |
+
gr.Markdown(' ### <a href="https://huggingface.co/spaces/panelforge/rainbow-media-real-v11" target="_blank" class="button-link">Try a more realistic model</a>') # Update links
|
70 |
|
71 |
result = gr.Image(label="Result", show_label=False)
|
72 |
|