Spaces:
Running
on
Zero
Running
on
Zero
- model_helper.py +5 -1
model_helper.py
CHANGED
|
@@ -1,5 +1,9 @@
|
|
| 1 |
# @title Model helper
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
from collections import Counter
|
| 4 |
import argparse
|
| 5 |
import torch
|
|
|
|
| 1 |
# @title Model helper
|
| 2 |
+
import sys
|
| 3 |
+
import os
|
| 4 |
+
current_directory = os.path.dirname(os.path.abspath(__file__)
|
| 5 |
+
sys.path.append(os.path.join(current_directory, 'src'))
|
| 6 |
+
|
| 7 |
from collections import Counter
|
| 8 |
import argparse
|
| 9 |
import torch
|