Gauri54damle commited on
Commit
ea19d4c
·
verified ·
1 Parent(s): cbda784

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ from diffusers import AutoencoderKL
10
 
11
 
12
  model = "stabilityai/stable-diffusion-xl-base-1.0"
13
- finetuningLayer = "Gauri54damle/sdxl-lora-McDBigMac-meal-model"
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  torch_dtype = torch.float16 if device.type == 'cuda' else torch.float32
@@ -111,7 +111,7 @@ with gr.Blocks(css=css) as demo:
111
  with gr.Column():
112
 
113
  model = gr.Dropdown(label="Base Model", value="stabilityai/stable-diffusion-xl-base-1.0")
114
- finetuningLayer= gr.Dropdown(label="Finetuning Layer", value="Gauri54damle/sdxl-lora-multi-object")
115
 
116
  prompt = gr.Textbox(label="Prompt", placeholder="photo of burger called McDBigMac placed on serving tray with fries called McDFries- it is unique identifier need to be used to identify burger")
117
 
 
10
 
11
 
12
  model = "stabilityai/stable-diffusion-xl-base-1.0"
13
+ finetuningLayer = "Gauri54damle/sdxl-lora-Fries-Coke-Burger-model"
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  torch_dtype = torch.float16 if device.type == 'cuda' else torch.float32
 
111
  with gr.Column():
112
 
113
  model = gr.Dropdown(label="Base Model", value="stabilityai/stable-diffusion-xl-base-1.0")
114
+ finetuningLayer= gr.Dropdown(label="Finetuning Layer", value="Gauri54damle/sdxl-lora-Fries-Coke-Burger-model")
115
 
116
  prompt = gr.Textbox(label="Prompt", placeholder="photo of burger called McDBigMac placed on serving tray with fries called McDFries- it is unique identifier need to be used to identify burger")
117