alexanz commited on
Commit
a85bcf5
·
verified ·
1 Parent(s): d472d55

default alexanz/SD14_lora_pusheen

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from diffusers import DiffusionPipeline, StableDiffusionPipeline
8
  import torch
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
- model_repo_id = "CompVis/stable-diffusion-v1-4" # Replace to the model you would like to use
12
 
13
  if torch.cuda.is_available():
14
  torch_dtype = torch.float16
@@ -97,7 +97,7 @@ with gr.Blocks(css=css) as demo:
97
  gr.Markdown(" # Text-to-Image Gradio Template")
98
  model_dropdown = gr.Dropdown(label="Model ID",
99
  choices=["alexanz/SD14_lora_pusheen", "CompVis/stable-diffusion-v1-4"],
100
- value="CompVis/stable-diffusion-v1-4")
101
  model_status = gr.Textbox(label="Model Status", interactive=False)
102
 
103
  with gr.Row():
 
8
  import torch
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
+ model_repo_id = "alexanz/SD14_lora_pusheen" # Replace to the model you would like to use
12
 
13
  if torch.cuda.is_available():
14
  torch_dtype = torch.float16
 
97
  gr.Markdown(" # Text-to-Image Gradio Template")
98
  model_dropdown = gr.Dropdown(label="Model ID",
99
  choices=["alexanz/SD14_lora_pusheen", "CompVis/stable-diffusion-v1-4"],
100
+ value="alexanz/SD14_lora_pusheen")
101
  model_status = gr.Textbox(label="Model Status", interactive=False)
102
 
103
  with gr.Row():