HAL1993 commited on
Commit
07ea755
·
verified ·
1 Parent(s): 0dfe7a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import gradio as gr
4
  client = Client("kratadata/Colabor-Image-Finetune")
5
 
6
  def generate_image(prompt, aspect_ratio):
7
- # Hardcode other parameters
8
- lora_model = "group1"
9
  num_inference_steps = 28
10
  guidance_scale = 3
11
  seed = -1
 
4
  client = Client("kratadata/Colabor-Image-Finetune")
5
 
6
  def generate_image(prompt, aspect_ratio):
7
+ # Change lora_model to a valid choice
8
+ lora_model = "pizza-ananas" # or "pizza-pilz" or "pizza-ananas-pilz"
9
  num_inference_steps = 28
10
  guidance_scale = 3
11
  seed = -1