Shivdutta commited on
Commit
f76de0a
·
verified ·
1 Parent(s): 9bc89fb

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -289,7 +289,7 @@ dict_styles = {
289
  }
290
  # dict_styles.keys()
291
 
292
- def inference(prompt, seed,style):
293
 
294
  if prompt is not None and style is not None and seed is not None:
295
  style = dict_styles[style]
@@ -304,11 +304,9 @@ description = "A simple Gradio interface to stylize Stable Diffusion outputs"
304
  examples = [['Pink Ferrari Car', 'Manga'], ['A man sipping wine wearing a spacesuit on the moon', 'GTA-5']] # Added valid styles
305
 
306
  demo = gr.Interface(inference,
307
- inputs = [gr.Textbox(label='Prompt'),
 
308
  gr.Textbox(label='Seed', value='24041975'),
309
- gr.Dropdown(['Dr Strange', 'GTA-5',
310
- 'Manga', 'Pokemon'], label='Style')
311
- ],
312
  outputs = [
313
  gr.Image(label="Stable Diffusion Output"),
314
  ],
 
289
  }
290
  # dict_styles.keys()
291
 
292
+ def inference(prompt, style,seed):
293
 
294
  if prompt is not None and style is not None and seed is not None:
295
  style = dict_styles[style]
 
304
  examples = [['Pink Ferrari Car', 'Manga'], ['A man sipping wine wearing a spacesuit on the moon', 'GTA-5']] # Added valid styles
305
 
306
  demo = gr.Interface(inference,
307
+ inputs = [gr.Textbox(label='Prompt'),
308
+ gr.Dropdown(['Dr Strange', 'GTA-5', 'Manga', 'Pokemon'], label='Style')],
309
  gr.Textbox(label='Seed', value='24041975'),
 
 
 
310
  outputs = [
311
  gr.Image(label="Stable Diffusion Output"),
312
  ],