ejschwartz commited on
Commit
5767a00
·
1 Parent(s): b3f7be6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ example = """int __fastcall sub_B0D04(int a1, int a2)
29
  return result;
30
  }"""
31
 
32
- examples = [j["input"] for j in json.load(open("gpt4_juiced_dataset.json"))]
33
 
34
  # Then create the pipeline with the model and tokenizer
35
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
 
29
  return result;
30
  }"""
31
 
32
+ examples = [json.loads(line)["input"] for line in open("gpt4_juiced_dataset.json", "r")]
33
 
34
  # Then create the pipeline with the model and tokenizer
35
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)