update

main.py

@@ -23,7 +23,12 @@ def compile(compiler, flags, source):
         temp_o_file_name = temp_o_file.name
 
     # Compile the C file to an object file
-    subprocess.run([compiler, "-c", temp_c_file_name] + shlex.split(flags) + ["-o", temp_o_file_name], check=True)
+    subprocess.run(
+        [compiler, "-c", temp_c_file_name]
+        + shlex.split(flags)
+        + ["-o", temp_o_file_name],
+        check=True,
+    )
 
     # Create a temporary file for the raw bytes
     with tempfile.NamedTemporaryFile(suffix=".raw", delete=True) as raw_bytes_file:
@@ -45,8 +50,9 @@ def compile(compiler, flags, source):
 
 
 def predict(target, source, compiler, flags):
+    target = bytes.fromhex(target)
     bytes = compile(compiler, flags, source)
-    return hexdump(bytes, result="return"), 1.0
+    return hexdump(bytes, result="return"), hexdump(target, result="return"), 1.0
 
 
 def run():
@@ -59,7 +65,11 @@ def run():
             gr.Textbox(label="Compiler", value="gcc"),
             gr.Textbox(label="Compiler Flags", value="-O2"),
         ],
-        outputs=[gr.Textbox(label="Compiled bytes"), gr.Number(label="Accuracy (TBD)")],
+        outputs=[
+            gr.Textbox(label="Compiled bytes"),
+            gr.Textbox(label="Target bytes"),
+            gr.Number(label="Accuracy (TBD)"),
+        ],
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)