Spaces:

ejschwartz
/

aidapal-space

Sleeping

ejschwartz commited on Apr 17

Commit

9f58c2e

1 Parent(s): 33dfc83

so close

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,8 +72,8 @@ def predict(code):
     raw_output = pipe_out[0]["generated_text"]
     output = raw_output
-    if output.startswith("``` json\n"):
-        output = output[9:]
     json_output = json.dumps([])
     try:

     raw_output = pipe_out[0]["generated_text"]
     output = raw_output
+    if output.startswith("```json\n"):
+        output = output[8:]
     json_output = json.dumps([])
     try: