zaidmehdi commited on
Commit
45c84cc
1 Parent(s): 37aa982

.utils import

Browse files
Files changed (1) hide show
  1. 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