Update api_v2.py
Browse files
api_v2.py
CHANGED
|
@@ -101,6 +101,7 @@ import os
|
|
| 101 |
import sys
|
| 102 |
import traceback
|
| 103 |
from typing import Generator
|
|
|
|
| 104 |
|
| 105 |
now_dir = os.getcwd()
|
| 106 |
sys.path.append(now_dir)
|
|
|
|
| 101 |
import sys
|
| 102 |
import traceback
|
| 103 |
from typing import Generator
|
| 104 |
+
import torch
|
| 105 |
|
| 106 |
now_dir = os.getcwd()
|
| 107 |
sys.path.append(now_dir)
|