twarner commited on
Commit
5d7555b
·
1 Parent(s): 89103e0

Filter training metadata from output

Browse files
Files changed (1) hide show
  1. app.py +9 -0
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)