ruslanmv commited on
Commit
649181c
·
verified ·
1 Parent(s): ebb25b2

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +5 -4
src/app.py CHANGED
@@ -127,8 +127,8 @@ def generate(
127
  yield collected_answer # Yield initial part of answer
128
 
129
  elif reasoning_started and not answer_started:
130
- collected_reasoning = text # Accumulate reasoning tokens
131
- yield text # Stream reasoning tokens
132
 
133
  elif answer_started:
134
  collected_answer += text # Accumulate answer tokens
@@ -238,7 +238,7 @@ def display_vision_conversation(conversation):
238
  assistant_content = conversation[i+1]["content"]
239
  assistant_text_parts = []
240
  for item in assistant_content:
241
- if item["type"] == "text":
242
  assistant_text_parts.append(item["text"])
243
  assistant_msg = "\n".join(assistant_text_parts).strip()
244
  i += 2
@@ -321,8 +321,9 @@ with gr.Blocks(fill_height=True, css_paths=css_file_path, head_paths=head_file_p
321
  vision_temperature_slider = gr.Slider(minimum=0.0, maximum=2.0, value=VISION_TEMPERATURE, step=0.01, label="Vision Temperature", elem_classes=["gr_accordion_element"])
322
  vision_top_p_slider = gr.Slider(minimum=0.0, maximum=1.0, value=VISION_TOP_P, step=0.01, label="Vision Top p", elem_classes=["gr_accordion_element"])
323
  vision_top_k_slider = gr.Slider(minimum=0, maximum=100, value=VISION_TOP_K, step=1, label="Vision Top k", elem_classes=["gr_accordion_element"])
 
324
  vision_max_tokens_slider = gr.Slider(minimum=10, maximum=300, value=VISION_MAX_TOKENS, step=1, label="Vision Max Tokens", elem_classes=["gr_accordion_element"])
325
- send_button = gr.Button("Send Message")
326
  clear_button = gr.Button("Clear Chat")
327
 
328
  # Conversation state variables for each branch.
 
127
  yield collected_answer # Yield initial part of answer
128
 
129
  elif reasoning_started and not answer_started:
130
+ collected_reasoning = text # Accumulate reasoning tokens
131
+ yield text # Stream reasoning tokens
132
 
133
  elif answer_started:
134
  collected_answer += text # Accumulate answer tokens
 
238
  assistant_content = conversation[i+1]["content"]
239
  assistant_text_parts = []
240
  for item in assistant_content:
241
+ if item["type"] == "text":
242
  assistant_text_parts.append(item["text"])
243
  assistant_msg = "\n".join(assistant_text_parts).strip()
244
  i += 2
 
321
  vision_temperature_slider = gr.Slider(minimum=0.0, maximum=2.0, value=VISION_TEMPERATURE, step=0.01, label="Vision Temperature", elem_classes=["gr_accordion_element"])
322
  vision_top_p_slider = gr.Slider(minimum=0.0, maximum=1.0, value=VISION_TOP_P, step=0.01, label="Vision Top p", elem_classes=["gr_accordion_element"])
323
  vision_top_k_slider = gr.Slider(minimum=0, maximum=100, value=VISION_TOP_K, step=1, label="Vision Top k", elem_classes=["gr_accordion_element"])
324
+ vision_max_tokens_slider = gr.Slider(minimum=10, maximum=300, value=VISION_MAX_TOKENS, step=1,
325
  vision_max_tokens_slider = gr.Slider(minimum=10, maximum=300, value=VISION_MAX_TOKENS, step=1, label="Vision Max Tokens", elem_classes=["gr_accordion_element"])
326
+ send_button = gr.Button("Send Message")
327
  clear_button = gr.Button("Clear Chat")
328
 
329
  # Conversation state variables for each branch.