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)
|