Spaces:
Sleeping
Sleeping
.utils import
Browse files- src/main.py +1 -1
src/main.py
CHANGED
@@ -4,7 +4,7 @@ import pickle
|
|
4 |
from flask import Flask, request, jsonify
|
5 |
from transformers import AutoModel, AutoTokenizer
|
6 |
|
7 |
-
from utils import extract_hidden_state
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
|
|
4 |
from flask import Flask, request, jsonify
|
5 |
from transformers import AutoModel, AutoTokenizer
|
6 |
|
7 |
+
from .utils import extract_hidden_state
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|