ashishtanwer commited on
Commit
adfe545
1 Parent(s): 924289b

Update utils_app.py

Browse files
Files changed (1) hide show
  1. utils_app.py +1 -10
utils_app.py CHANGED
@@ -7,11 +7,6 @@ import gradio as gr
7
 
8
  keras.mixed_precision.set_global_policy("mixed_float16")
9
 
10
- keras_model_list = [
11
- "kadirnar/dreambooth_diffusion_model_v5",
12
- "kadirnar/dreambooth_diffusion_model_v3"
13
- ]
14
-
15
  stable_prompt_list = [
16
  "a photo of sshh shoe",
17
  ]
@@ -54,11 +49,7 @@ def keras_stable_diffusion_app():
54
  with gr.Blocks():
55
  with gr.Row():
56
  with gr.Column():
57
- keras_text2image_model_path = gr.Dropdown(
58
- choices=keras_model_list,
59
- value=keras_model_list[0],
60
- label='Text-Image Model Id'
61
- )
62
 
63
  keras_text2image_prompt = gr.Textbox(
64
  lines=1,
 
7
 
8
  keras.mixed_precision.set_global_policy("mixed_float16")
9
 
 
 
 
 
 
10
  stable_prompt_list = [
11
  "a photo of sshh shoe",
12
  ]
 
49
  with gr.Blocks():
50
  with gr.Row():
51
  with gr.Column():
52
+ keras_text2image_model_path = "ashishtanwer/shoe"
 
 
 
 
53
 
54
  keras_text2image_prompt = gr.Textbox(
55
  lines=1,