Try to remove markdown

README.md

@@ -19,4 +19,4 @@ names, when given the output of Hex-Rays decompiler output of a function.  More
 ## TODO / Issues
 
 * We currently use `transformers` which de-quantizes the gguf.  This is easy but inefficient.  Can we use llama.cpp or Ollama with zerogpu?
-* Model returns the markdown json prefix often.  Is this something I am doing wrong?
+* Model returns the markdown json prefix often.  Is this something I am doing wrong?  Currently we remove it in present to enable JSON parsing.

app.py

@@ -70,7 +70,10 @@ def predict(code):
     )
     print(f"Pipe out: {repr(pipe_out)}")
 
-    output = pipe_out[0]["generated_text"]
+    raw_output = pipe_out[0]["generated_text"]
+    output = raw_output
+    if output.startswith("``` json\n"):
+        output = output[9:]
 
     json_output = json.dumps([])
     try:
@@ -81,7 +84,7 @@ def predict(code):
 
     print(f"JSON output: {repr(json_output)}")
 
-    return json_output, output
+    return json_output, raw_output
 
 
 demo = gr.Interface(