rahul7star commited on
Commit
49fa7d4
·
verified ·
1 Parent(s): 5daa352

Update app_flash1.py

Browse files
Files changed (1) hide show
  1. 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")