Test JSON

app.py

@@ -72,7 +72,14 @@ def predict(code):
 
     output = pipe_out[0]["generated_text"]
 
-    return output, output
+    json_output = json.dumps("Failed")
+    try:
+        json.loads(output)
+        json_output = output
+    except Exception:
+        pass
+
+    return json_output, output
 
 
 demo = gr.Interface(