Spaces:

Chamin09
/

ai_agents_sustainable

Sleeping

Chamin09 commited on Mar 2

Commit

3251ecf

verified ·

1 Parent(s): 4745123

Update agents/image_processing_agent.py

Files changed (1) hide show

agents/image_processing_agent.py CHANGED Viewed

@@ -2,6 +2,7 @@
 import logging
 import os
 import gc
 from typing import Dict, List, Optional, Tuple, Union, Any
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed

 import logging
 import os
 import gc
+import torch
 from typing import Dict, List, Optional, Tuple, Union, Any
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed