Update app_flash1.py
Browse files- app_flash1.py +3 -1
app_flash1.py
CHANGED
|
@@ -191,6 +191,7 @@ def make_enhance_fn(model, tokenizer, embed_model, long_emb, dataset):
|
|
| 191 |
enhanced = dataset[best]["long_prompt"]
|
| 192 |
chat.append({"role": "user", "content": prompt})
|
| 193 |
chat.append({"role": "assistant", "content": enhanced})
|
|
|
|
| 194 |
return chat
|
| 195 |
return fn
|
| 196 |
|
|
@@ -201,7 +202,8 @@ def make_enhance_fn(model, tokenizer, embed_model, long_emb, dataset):
|
|
| 201 |
with gr.Blocks(title="✨ FlashPack Prompt Enhancer") as demo:
|
| 202 |
gr.Markdown("## 🧠 FlashPack Prompt Enhancer (CPU)\nShort → Long prompt expander")
|
| 203 |
|
| 204 |
-
chatbot = gr.Chatbot(height=400)
|
|
|
|
| 205 |
user_input = gr.Textbox(label="Your prompt")
|
| 206 |
send_btn = gr.Button("🚀 Enhance Prompt", variant="primary")
|
| 207 |
clear_btn = gr.Button("🧹 Clear")
|
|
|
|
| 191 |
enhanced = dataset[best]["long_prompt"]
|
| 192 |
chat.append({"role": "user", "content": prompt})
|
| 193 |
chat.append({"role": "assistant", "content": enhanced})
|
| 194 |
+
|
| 195 |
return chat
|
| 196 |
return fn
|
| 197 |
|
|
|
|
| 202 |
with gr.Blocks(title="✨ FlashPack Prompt Enhancer") as demo:
|
| 203 |
gr.Markdown("## 🧠 FlashPack Prompt Enhancer (CPU)\nShort → Long prompt expander")
|
| 204 |
|
| 205 |
+
chatbot = gr.Chatbot(height=400, type="messages")
|
| 206 |
+
|
| 207 |
user_input = gr.Textbox(label="Your prompt")
|
| 208 |
send_btn = gr.Button("🚀 Enhance Prompt", variant="primary")
|
| 209 |
clear_btn = gr.Button("🧹 Clear")
|