Yaron Koresh commited on
Commit
77ffbf9
·
verified ·
1 Parent(s): ebe916c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -77,14 +77,14 @@ def generate_random_string(length):
77
  characters = string.ascii_letters + string.digits
78
  return ''.join(random.choice(characters) for _ in range(length))
79
 
80
- @spaces.GPU(duration=50)
81
  def Piper(_do):
82
  try:
83
  retu = pipe(
84
  _do,
85
- height=512,
86
- width=768,
87
- num_inference_steps=50,
88
  max_sequence_length=512,
89
  guidance_scale=2
90
  )
@@ -95,9 +95,9 @@ def Piper(_do):
95
  def infer(prompt_en):
96
  name = generate_random_string(12)+".png"
97
  if prompt_en == "":
98
- _do = 'photographed realistic scene'
99
  else:
100
- _do = f'photographed { prompt_en } realistic scene'
101
  output = Piper(_do)
102
  if output == "":
103
  return output
@@ -126,7 +126,7 @@ footer {
126
  max-width: 15cm !important;
127
  }
128
  .image-container {
129
- aspect-ratio: 768 / 512 !important;
130
  }
131
  .dropdown-arrow {
132
  display: none !important;
@@ -161,7 +161,6 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
161
  with gr.Row():
162
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
163
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
164
- result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
165
 
166
  def _ret(idx,p1):
167
  print(f'Starting {idx}: {p1}')
 
77
  characters = string.ascii_letters + string.digits
78
  return ''.join(random.choice(characters) for _ in range(length))
79
 
80
+ @spaces.GPU(duration=35)
81
  def Piper(_do):
82
  try:
83
  retu = pipe(
84
  _do,
85
+ height=480,
86
+ width=480,
87
+ num_inference_steps=15,
88
  max_sequence_length=512,
89
  guidance_scale=2
90
  )
 
95
  def infer(prompt_en):
96
  name = generate_random_string(12)+".png"
97
  if prompt_en == "":
98
+ _do = 'photographed scene, highly detailed, vivid colors, realistic content, dynamic poze, deep field'
99
  else:
100
+ _do = f'{ prompt_en } photograph, highly detailed, vivid colors, realistic content, dynamic poze, deep field'
101
  output = Piper(_do)
102
  if output == "":
103
  return output
 
126
  max-width: 15cm !important;
127
  }
128
  .image-container {
129
+ aspect-ratio: 512 / 512 !important;
130
  }
131
  .dropdown-arrow {
132
  display: none !important;
 
161
  with gr.Row():
162
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
163
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
164
 
165
  def _ret(idx,p1):
166
  print(f'Starting {idx}: {p1}')