prithivMLmods commited on
Commit
a4c548f
·
verified ·
1 Parent(s): 3a454bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -6,13 +6,7 @@ pipe = pipeline('text-generation', model='daspartho/prompt-extend')
6
  def extend_prompt(prompt):
7
  return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]
8
 
9
- examples = [
10
- ['a carnival of robots in a neon cityscape'],
11
- ["a mermaid playing electric guitar underwater"],
12
- ['a swarm of fireflies illuminating a moonlit forest'],
13
- ["a time-traveling pirate ship sailing through a storm of stars"],
14
- ["a mystical portal hidden in a waterfall"],
15
- ]
16
 
17
  iface = gr.Interface(
18
  description = "Enter a main idea for a prompt, and the model will attempt to add suitable style cues.",
@@ -20,7 +14,6 @@ iface = gr.Interface(
20
  fn=extend_prompt,
21
  inputs=gr.Text(label="Type the prompt here"),
22
  outputs=gr.TextArea(label='Extended prompt'),
23
- examples=examples,
24
  title="Prompt Extender"
25
  )
26
 
 
6
  def extend_prompt(prompt):
7
  return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]
8
 
9
+
 
 
 
 
 
 
10
 
11
  iface = gr.Interface(
12
  description = "Enter a main idea for a prompt, and the model will attempt to add suitable style cues.",
 
14
  fn=extend_prompt,
15
  inputs=gr.Text(label="Type the prompt here"),
16
  outputs=gr.TextArea(label='Extended prompt'),
 
17
  title="Prompt Extender"
18
  )
19