Upload cogagent_infer.py with huggingface_hub
Browse files- cogagent_infer.py +1 -1
cogagent_infer.py
CHANGED
|
@@ -199,7 +199,7 @@ def main():
|
|
| 199 |
res.append(ans)
|
| 200 |
round_num += 1
|
| 201 |
print(res)
|
| 202 |
-
with open(
|
| 203 |
print("Writing to json file")
|
| 204 |
json.dump(res, file, indent=4)
|
| 205 |
|
|
|
|
| 199 |
res.append(ans)
|
| 200 |
round_num += 1
|
| 201 |
print(res)
|
| 202 |
+
with open(args.output_json, "w") as file:
|
| 203 |
print("Writing to json file")
|
| 204 |
json.dump(res, file, indent=4)
|
| 205 |
|