Filter training metadata from output
Browse files
app.py
CHANGED
|
@@ -528,6 +528,15 @@ def generate(prompt: str, temperature: float, max_tokens: int, num_steps: int, g
|
|
| 528 |
if is_v3:
|
| 529 |
gcode = gcode.replace("<newline>", "\n")
|
| 530 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 531 |
print(f"Decoded gcode length: {len(gcode)} chars")
|
| 532 |
|
| 533 |
gcode = validate_gcode(gcode)
|
|
|
|
| 528 |
if is_v3:
|
| 529 |
gcode = gcode.replace("<newline>", "\n")
|
| 530 |
|
| 531 |
+
# Filter out training metadata lines
|
| 532 |
+
filtered_lines = []
|
| 533 |
+
for line in gcode.split("\n"):
|
| 534 |
+
# Skip metadata headers from training data
|
| 535 |
+
if line.startswith("Source:") or line.startswith(";Generated"):
|
| 536 |
+
continue
|
| 537 |
+
filtered_lines.append(line)
|
| 538 |
+
gcode = "\n".join(filtered_lines)
|
| 539 |
+
|
| 540 |
print(f"Decoded gcode length: {len(gcode)} chars")
|
| 541 |
|
| 542 |
gcode = validate_gcode(gcode)
|