broadfield commited on
Commit
d90f5ed
·
verified ·
1 Parent(s): d9c7091

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -243,7 +243,7 @@ def agent(prompt_in,history,mod=2):
243
  in_data[1]=com
244
  out_o =generate(prompt,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
245
  out_w=list(out_o)
246
- ret1,ret2 = parse_json(out_w[2])
247
  build_space(out_w[0],ret1,ret2)
248
  elif 'IMAGE' in fn:
249
  print('IMAGE called')
 
243
  in_data[1]=com
244
  out_o =generate(prompt,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
245
  out_w=list(out_o)
246
+ ret1,ret2 = parse_json(out_w[2].split('<|im_end|>')[0])
247
  build_space(out_w[0],ret1,ret2)
248
  elif 'IMAGE' in fn:
249
  print('IMAGE called')