LDY commited on
Commit
8a3ba61
1 Parent(s): 64aee06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -23
app.py CHANGED
@@ -28,7 +28,7 @@ model = MinDalle(
28
  dtype=torch.float32
29
  )
30
 
31
- log_gpu_memory()
32
 
33
  def run_model(
34
  text: str,
@@ -83,7 +83,7 @@ def run_model(
83
  is_verbose = True
84
  )
85
 
86
- log_gpu_memory()
87
 
88
  ext = 'png' if bool(save_as_png) else 'jpg'
89
  filename = filename_from_text(text)
@@ -159,22 +159,8 @@ with demo:
159
 
160
  gradio.Examples(
161
  examples=[
162
- ['Rusty Iron Man suit found abandoned in the woods being reclaimed by nature', 3, 'examples/rusty-iron-man.jpg'],
163
- ['Moai statue giving a TED Talk', 5, 'examples/moai-statue.jpg'],
164
- ['Court sketch of Godzilla on trial', 5, 'examples/godzilla-trial.jpg'],
165
- ['lofi nuclear war to relax and study to', 5, 'examples/lofi-nuclear-war.jpg'],
166
- ['Karl Marx slimed at Kids Choice Awards', 4, 'examples/marx-slimed.jpg'],
167
- ['Scientists trying to rhyme orange with banana', 4, 'examples/scientists-rhyme.jpg'],
168
- ['Jesus turning water into wine on Americas Got Talent', 5, 'examples/jesus-talent.jpg'],
169
- ['Elmo in a street riot throwing a Molotov cocktail, hyperrealistic', 5, 'examples/elmo-riot.jpg'],
170
- ['Trail cam footage of gollum eating watermelon', 4, 'examples/gollum.jpg'],
171
- ['Funeral at Whole Foods', 4, 'examples/funeral-whole-foods.jpg'],
172
- ['Singularity, hyperrealism', 5, 'examples/singularity.jpg'],
173
- ['Astronaut riding a horse hyperrealistic', 5, 'examples/astronaut-horse.jpg'],
174
- ['An astronaut walking on Mars next to a Starship rocket, realistic', 5, 'examples/astronaut-mars.jpg'],
175
- ['Nuclear explosion broccoli', 4, 'examples/nuclear-broccoli.jpg'],
176
- ['Dali painting of WALL·E', 5, 'examples/dali-walle.jpg'],
177
- ['Cleopatra checking her iPhone', 4, 'examples/cleopatra-iphone.jpg'],
178
  ],
179
  inputs=[
180
  input_text,
@@ -200,10 +186,5 @@ with demo:
200
  ]
201
  )
202
 
203
- gradio.Markdown(
204
- """
205
- ### **[❤️ Sponsor](https://github.com/sponsors/kuprel)**
206
- """
207
- )
208
 
209
  demo.launch()
 
28
  dtype=torch.float32
29
  )
30
 
31
+ ## log_gpu_memory()
32
 
33
  def run_model(
34
  text: str,
 
83
  is_verbose = True
84
  )
85
 
86
+ ## log_gpu_memory()
87
 
88
  ext = 'png' if bool(save_as_png) else 'jpg'
89
  filename = filename_from_text(text)
 
159
 
160
  gradio.Examples(
161
  examples=[
162
+ ['A white cat with golden sunglasses on, pink background, studio lighting, 4k, award winning photography', 2, 'examples/cat.png'],
163
+ ['an astronaut dancing on the moon’s surface, close-up photo', 2, 'examples/astronaut.png'],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
164
  ],
165
  inputs=[
166
  input_text,
 
186
  ]
187
  )
188
 
 
 
 
 
 
189
 
190
  demo.launch()