Spaces:
Build error
Build error
UI changes
Browse files
app.py
CHANGED
@@ -32,8 +32,13 @@ css = """
|
|
32 |
#share-btn-container .wrap {
|
33 |
display: none !important;
|
34 |
}
|
|
|
35 |
"""
|
36 |
|
|
|
|
|
|
|
|
|
37 |
# Download model from HF Hub.
|
38 |
ckpt_path = huggingface_hub.hf_hub_download(
|
39 |
repo_id='jykoh/fromage', filename='pretrained_ckpt.pth.tar')
|
@@ -141,8 +146,8 @@ with gr.Blocks(css=css) as demo:
|
|
141 |
For faster inference without waiting in queue, you may duplicate the space and use your own GPU. <a href="https://huggingface.co/spaces/haoheliu/audioldm-text-to-audio-generation?duplicate=true"><img style="margin-top: 0em; margin-bottom: 0em" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>
|
142 |
""")
|
143 |
|
144 |
-
chatbot = gr.Chatbot(elem_id="chatbot")
|
145 |
gr_state = gr.State([[], [], None]) # chat_history, input_image
|
|
|
146 |
with gr.Row(visible=False) as share_group:
|
147 |
with gr.Group(elem_id="share-btn-container"):
|
148 |
community_icon = gr.HTML(community_icon_html)
|
@@ -150,38 +155,40 @@ with gr.Blocks(css=css) as demo:
|
|
150 |
share_button = gr.Button("Share to community", elem_id="share-btn")
|
151 |
|
152 |
with gr.Row():
|
153 |
-
with gr.Column(scale=0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
154 |
ret_scale_factor = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.1, interactive=True,
|
155 |
-
label="
|
156 |
max_ret_images = gr.Number(
|
157 |
minimum=0, maximum=3, value=1, precision=1, interactive=True, label="Max images to return")
|
158 |
gr_max_len = gr.Slider(minimum=1, maximum=64, value=32,
|
159 |
step=1, interactive=True, label="Max # of words returned")
|
160 |
gr_temperature = gr.Slider(
|
161 |
-
minimum=0.0, maximum=1.0, value=0.0, interactive=True, label="Temperature")
|
162 |
-
|
163 |
-
with gr.Column(scale=0.7, min_width=400):
|
164 |
-
image_btn = gr.UploadButton("🖼️ Image Input", file_types=["image"])
|
165 |
-
text_input = gr.Textbox(label="Type a message", lines=1, placeholder="Upload an image above [optional]. Then enter a text prompt, and press enter!")
|
166 |
|
167 |
-
|
168 |
-
|
169 |
-
|
170 |
-
|
171 |
-
with gr.Column(scale=0.33):
|
172 |
-
clear_last_btn = gr.Button("Clear Last Round")
|
173 |
-
with gr.Column(scale=0.33):
|
174 |
-
clear_btn = gr.Button("Clear All")
|
175 |
|
176 |
text_input.submit(generate_for_prompt, [text_input, gr_state, ret_scale_factor,
|
177 |
-
max_ret_images, gr_max_len, gr_temperature], [gr_state, chatbot, share_group]
|
178 |
-
)
|
179 |
-
# _js = "() => document.getElementById('#chatbot').scrollTop = document.getElementById('#chatbot').scrollHeight")
|
180 |
text_input.submit(lambda: "", None, text_input) # Reset chatbox.
|
181 |
submit_btn.click(generate_for_prompt, [text_input, gr_state, ret_scale_factor,
|
182 |
-
max_ret_images, gr_max_len, gr_temperature], [gr_state, chatbot, share_group]
|
183 |
-
)
|
184 |
-
# _js = "() => document.getElementById('#chatbot').scrollTop = document.getElementById('#chatbot').scrollHeight")
|
185 |
submit_btn.click(lambda: "", None, text_input) # Reset chatbox.
|
186 |
|
187 |
image_btn.upload(upload_image, [gr_state, image_btn], [gr_state, chatbot])
|
|
|
32 |
#share-btn-container .wrap {
|
33 |
display: none !important;
|
34 |
}
|
35 |
+
#chatbot { height: 300px; }
|
36 |
"""
|
37 |
|
38 |
+
examples = [
|
39 |
+
# [[[], [], None], [("Show me some pictures in action?"), ('OK')]],
|
40 |
+
]
|
41 |
+
|
42 |
# Download model from HF Hub.
|
43 |
ckpt_path = huggingface_hub.hf_hub_download(
|
44 |
repo_id='jykoh/fromage', filename='pretrained_ckpt.pth.tar')
|
|
|
146 |
For faster inference without waiting in queue, you may duplicate the space and use your own GPU. <a href="https://huggingface.co/spaces/haoheliu/audioldm-text-to-audio-generation?duplicate=true"><img style="margin-top: 0em; margin-bottom: 0em" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>
|
147 |
""")
|
148 |
|
|
|
149 |
gr_state = gr.State([[], [], None]) # chat_history, input_image
|
150 |
+
|
151 |
with gr.Row(visible=False) as share_group:
|
152 |
with gr.Group(elem_id="share-btn-container"):
|
153 |
community_icon = gr.HTML(community_icon_html)
|
|
|
155 |
share_button = gr.Button("Share to community", elem_id="share-btn")
|
156 |
|
157 |
with gr.Row():
|
158 |
+
with gr.Column(scale=0.7, min_width=500):
|
159 |
+
with gr.Row():
|
160 |
+
chatbot = gr.Chatbot(elem_id="chatbot", label="FROMAGe Chatbot")
|
161 |
+
with gr.Row():
|
162 |
+
image_btn = gr.UploadButton("🖼️ Upload Image", file_types=["image"])
|
163 |
+
|
164 |
+
text_input = gr.Textbox(lines=4, label="Message", placeholder="Type a message")
|
165 |
+
|
166 |
+
with gr.Column():
|
167 |
+
submit_btn = gr.Button(
|
168 |
+
"Submit", interactive=True, variant="primary")
|
169 |
+
clear_last_btn = gr.Button("Undo")
|
170 |
+
clear_btn = gr.Button("Reset All")
|
171 |
+
|
172 |
+
with gr.Column(scale=0.3, min_width=200):
|
173 |
ret_scale_factor = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.1, interactive=True,
|
174 |
+
label="Frequency multiplier for returning images (higher means more frequent)")
|
175 |
max_ret_images = gr.Number(
|
176 |
minimum=0, maximum=3, value=1, precision=1, interactive=True, label="Max images to return")
|
177 |
gr_max_len = gr.Slider(minimum=1, maximum=64, value=32,
|
178 |
step=1, interactive=True, label="Max # of words returned")
|
179 |
gr_temperature = gr.Slider(
|
180 |
+
minimum=0.0, maximum=1.0, value=0.0, interactive=True, label="Temperature (0 for deterministic, higher for more randomness)")
|
|
|
|
|
|
|
|
|
181 |
|
182 |
+
examples = gr.Examples(
|
183 |
+
examples=examples,
|
184 |
+
inputs=[gr_state, chatbot],
|
185 |
+
)
|
|
|
|
|
|
|
|
|
186 |
|
187 |
text_input.submit(generate_for_prompt, [text_input, gr_state, ret_scale_factor,
|
188 |
+
max_ret_images, gr_max_len, gr_temperature], [gr_state, chatbot, share_group])
|
|
|
|
|
189 |
text_input.submit(lambda: "", None, text_input) # Reset chatbox.
|
190 |
submit_btn.click(generate_for_prompt, [text_input, gr_state, ret_scale_factor,
|
191 |
+
max_ret_images, gr_max_len, gr_temperature], [gr_state, chatbot, share_group])
|
|
|
|
|
192 |
submit_btn.click(lambda: "", None, text_input) # Reset chatbox.
|
193 |
|
194 |
image_btn.upload(upload_image, [gr_state, image_btn], [gr_state, chatbot])
|