Spaces:
Running
Running
broadfield
commited on
Update app.py
Browse files
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')
|