Commit
•
3d9bffc
1
Parent(s):
124a3b5
Update app_scribble.py
Browse files- app_scribble.py +21 -5
app_scribble.py
CHANGED
@@ -6,17 +6,28 @@ from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
|
|
6 |
MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
|
7 |
from utils import randomize_seed_fn
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
def create_demo(process):
|
11 |
with gr.Blocks() as demo:
|
12 |
with gr.Row():
|
13 |
with gr.Column():
|
14 |
image = gr.Image()
|
15 |
-
|
16 |
-
|
17 |
-
prompt = gr.Textbox(label='Prompt')
|
18 |
-
with gr.Column():
|
19 |
-
prompt_btn = gr.Button("Button")
|
20 |
preprocessor_name = gr.Radio(
|
21 |
label='Preprocessor',
|
22 |
choices=['HED', 'PidiNet', 'None'],
|
@@ -96,6 +107,11 @@ def create_demo(process):
|
|
96 |
outputs=result,
|
97 |
api_name=False,
|
98 |
)
|
|
|
|
|
|
|
|
|
|
|
99 |
run_button.click(
|
100 |
fn=randomize_seed_fn,
|
101 |
inputs=[seed, randomize_seed],
|
|
|
6 |
MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
|
7 |
from utils import randomize_seed_fn
|
8 |
|
9 |
+
def magic_prompt(prompt):
|
10 |
+
openai.api_key = "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"
|
11 |
+
|
12 |
+
completion = openai.ChatCompletion.create(
|
13 |
+
model="gpt-3.5-turbo",
|
14 |
+
messages=[
|
15 |
+
{"role": "system", "content": "You are a helpful assistant."},
|
16 |
+
{"role": "user", "content": "Hello!"}
|
17 |
+
]
|
18 |
+
)
|
19 |
+
|
20 |
+
mag_prompt = str(completion.choices[0].message.content)
|
21 |
+
return mag_prompt
|
22 |
+
|
23 |
|
24 |
def create_demo(process):
|
25 |
with gr.Blocks() as demo:
|
26 |
with gr.Row():
|
27 |
with gr.Column():
|
28 |
image = gr.Image()
|
29 |
+
prompt = gr.Textbox(label='Prompt')
|
30 |
+
mag_prompt_btn = gr.Button("✨Magic Prompt")
|
|
|
|
|
|
|
31 |
preprocessor_name = gr.Radio(
|
32 |
label='Preprocessor',
|
33 |
choices=['HED', 'PidiNet', 'None'],
|
|
|
107 |
outputs=result,
|
108 |
api_name=False,
|
109 |
)
|
110 |
+
mag_prompt_btn.click(
|
111 |
+
fn=magic_prompt,
|
112 |
+
inputs=prompt,
|
113 |
+
outputs=prompt,
|
114 |
+
)
|
115 |
run_button.click(
|
116 |
fn=randomize_seed_fn,
|
117 |
inputs=[seed, randomize_seed],
|