chansung commited on
Commit
1db9851
1 Parent(s): 23c3fd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -121,6 +121,15 @@ class GradioLLaMA2ChatPPManager(LLaMA2ChatPPManager):
121
 
122
  TOKEN = os.getenv('HF_TOKEN')
123
  MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
 
 
 
 
 
 
 
 
 
124
 
125
  pipe = StableDiffusionPipeline.from_pretrained(
126
  "nota-ai/bk-sdm-base", torch_dtype=torch.float16
@@ -509,7 +518,7 @@ with gr.Blocks(css=STYLES) as demo:
509
  with gr.Tab("Control"):
510
  with gr.Column(elem_classes=['group-border']):
511
  gr.Markdown('### title')
512
- title = gr.Textbox("pokemon training story", elem_classes=['no-label'])
513
 
514
  with gr.Column(elem_classes=['group-border']):
515
  with gr.Column():
 
121
 
122
  TOKEN = os.getenv('HF_TOKEN')
123
  MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
124
+ TITLE_PLACEHOLDERS = [
125
+ 'Pokemon training story',
126
+ 'The Sun of Shangri-La',
127
+ 'Man In The Future',
128
+ 'Friends',
129
+ 'Cyborg Of A Beast',
130
+ 'Man At The Graveyard',
131
+ 'Vampire Of The Land'
132
+ ]
133
 
134
  pipe = StableDiffusionPipeline.from_pretrained(
135
  "nota-ai/bk-sdm-base", torch_dtype=torch.float16
 
518
  with gr.Tab("Control"):
519
  with gr.Column(elem_classes=['group-border']):
520
  gr.Markdown('### title')
521
+ title = gr.Textbox(random.choice(TITLE_PLACEHOLDERS), elem_classes=['no-label'])
522
 
523
  with gr.Column(elem_classes=['group-border']):
524
  with gr.Column():