Upload cogagent_infer.py with huggingface_hub
Browse files- cogagent_infer.py +3 -2
cogagent_infer.py
CHANGED
|
@@ -199,8 +199,9 @@ def main():
|
|
| 199 |
res.append(ans)
|
| 200 |
round_num += 1
|
| 201 |
print(res)
|
| 202 |
-
with open("args.output_json", "w"
|
| 203 |
-
|
|
|
|
| 204 |
|
| 205 |
|
| 206 |
|
|
|
|
| 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 |
|
| 206 |
|
| 207 |
|