kadirnar commited on
Commit
e9fe2c3
1 Parent(s): 40ac53b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -7,15 +7,6 @@ os.environ["CUDA_VISIBLE_DEVICES"]="0"
7
  import torch
8
  import gradio as gr
9
 
10
- best_model_list = [
11
- "stabilityai/stable-diffusion-2-inpainting",
12
- "runwayml/stable-diffusion-v1-5",
13
- "CompVis/stable-diffusion-v1-4",
14
- "prompthero/openjourney",
15
- "dreamlike-art/dreamlike-photoreal-2.0",
16
- "dreamlike-art/dreamlike-diffusion-1.0",
17
- ]
18
-
19
  orig_prompt = "Create a relaxing atmosphere with the use of plants and other natural elements. Such as a hanging terrarium or a wall-mounted planter. Include plenty of storage options to keep the space organized and clutter-free. Consider adding a vanity with double sinks and plenty of drawers and cabinets. As well as a wall mounted medicine and towel storage."
20
  orig_negative_prompt = "lurry, bad art, blurred, text, watermark"
21
 
@@ -70,9 +61,8 @@ def stable_diffusion_zoom_out(
70
  write_video(save_path, all_frames, fps=fps)
71
  return save_path
72
 
73
-
74
  inputs = [
75
- gr.Dropdown(choices=best_model_list,default="stabilityai/stable-diffusion-2-inpainting", label="Model"),
76
  gr.inputs.Textbox(lines=5, default=orig_prompt, label="Prompt"),
77
  gr.inputs.Textbox(lines=1, default=orig_negative_prompt, label="Negative Prompt"),
78
  gr.inputs.Slider(minimum=1, maximum=64, default=32, label="Steps"),
 
7
  import torch
8
  import gradio as gr
9
 
 
 
 
 
 
 
 
 
 
10
  orig_prompt = "Create a relaxing atmosphere with the use of plants and other natural elements. Such as a hanging terrarium or a wall-mounted planter. Include plenty of storage options to keep the space organized and clutter-free. Consider adding a vanity with double sinks and plenty of drawers and cabinets. As well as a wall mounted medicine and towel storage."
11
  orig_negative_prompt = "lurry, bad art, blurred, text, watermark"
12
 
 
61
  write_video(save_path, all_frames, fps=fps)
62
  return save_path
63
 
 
64
  inputs = [
65
+ gr.inputs.Textbox(lines=1, default="stabilityai/stable-diffusion-2-inpainting", label="Model"),
66
  gr.inputs.Textbox(lines=5, default=orig_prompt, label="Prompt"),
67
  gr.inputs.Textbox(lines=1, default=orig_negative_prompt, label="Negative Prompt"),
68
  gr.inputs.Slider(minimum=1, maximum=64, default=32, label="Steps"),