LukasHug commited on
Commit
2b4048a
·
verified ·
1 Parent(s): 422fbaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -134,9 +134,9 @@ default_conversation = SimpleConversation()
134
 
135
 
136
  def wrap_taxonomy(text):
137
- """Wraps user input with taxonomy if not already present"""
138
- if policy_v1 not in text:
139
- return policy_v1 + "\n\n" + text
140
  return text
141
 
142
 
@@ -326,7 +326,7 @@ def add_text(state, text, image, image_process_mode, request: gr.Request):
326
  state.set_prompt(prompt=prompt, image=image)
327
  state.skip_next = False
328
 
329
- return (state, state.to_gradio_chatbot(), default_taxonomy, None) + (disable_btn,) * 5
330
 
331
 
332
  def llava_bot(state, temperature, top_p, max_new_tokens, request: gr.Request):
@@ -614,4 +614,4 @@ demo.queue(
614
  server_name=args.host,
615
  server_port=args.port,
616
  share=args.share
617
- )
 
134
 
135
 
136
  def wrap_taxonomy(text):
137
+ """Return user-provided taxonomy unless the input is empty."""
138
+ if text is None or not text.strip():
139
+ return policy_v1
140
  return text
141
 
142
 
 
326
  state.set_prompt(prompt=prompt, image=image)
327
  state.skip_next = False
328
 
329
+ return (state, state.to_gradio_chatbot(), text, None) + (disable_btn,) * 5
330
 
331
 
332
  def llava_bot(state, temperature, top_p, max_new_tokens, request: gr.Request):
 
614
  server_name=args.host,
615
  server_port=args.port,
616
  share=args.share
617
+ )