Spaces:
Sleeping
Sleeping
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')
|