panelforge commited on
Commit
70fb10b
·
verified ·
1 Parent(s): 20d6cec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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-v10-sdxl" # Replace to the model you would like to use
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://example.com" target="_blank" class="button-link">Try a more realistic model</a>')
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