Spaces:
Sleeping
Sleeping
Update src/app.py
Browse files- 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 |
-
|
| 131 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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.
|