Update inference.py
Browse files- inference.py +5 -0
inference.py
CHANGED
|
@@ -1,5 +1,10 @@
|
|
| 1 |
import argparse
|
|
|
|
| 2 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
from typing import Dict, Optional, Tuple, List
|
| 4 |
from omegaconf import OmegaConf
|
| 5 |
from PIL import Image
|
|
|
|
| 1 |
import argparse
|
| 2 |
+
import sys
|
| 3 |
import os
|
| 4 |
+
|
| 5 |
+
# Add the current directory to Python path
|
| 6 |
+
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
| 7 |
+
|
| 8 |
from typing import Dict, Optional, Tuple, List
|
| 9 |
from omegaconf import OmegaConf
|
| 10 |
from PIL import Image
|