Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -206,7 +206,7 @@ def model_inference(input_dict, history):
|
|
206 |
|
207 |
# Process the full segment (e.g., remove <|im_end|>)
|
208 |
processed_segment = current_model_output_segment.split("<|im_end|>", 1)[0] if "<|im_end|>" in current_model_output_segment else current_model_output_segment
|
209 |
-
messages.append(dict(role='assistant', content=processed_segment))
|
210 |
# Append this processed segment to the cumulative display string for Gradio
|
211 |
complete_assistant_response_for_gradio += [processed_segment + "\n\n"]
|
212 |
yield complete_assistant_response_for_gradio # Ensure the fully processed segment is yielded to Gradio
|
|
|
206 |
|
207 |
# Process the full segment (e.g., remove <|im_end|>)
|
208 |
processed_segment = current_model_output_segment.split("<|im_end|>", 1)[0] if "<|im_end|>" in current_model_output_segment else current_model_output_segment
|
209 |
+
# messages.append(dict(role='assistant', content=processed_segment))
|
210 |
# Append this processed segment to the cumulative display string for Gradio
|
211 |
complete_assistant_response_for_gradio += [processed_segment + "\n\n"]
|
212 |
yield complete_assistant_response_for_gradio # Ensure the fully processed segment is yielded to Gradio
|