Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ def process_file(vendor_name, mode, file):
|
|
| 102 |
vendor_scores = {}
|
| 103 |
for csv_file, vector_file, score_file in zip(csv_files, vector_files, score_files):
|
| 104 |
with open(os.path.join('data', vector_file), 'rb') as f:
|
| 105 |
-
vendor_name = os.path.splitext(vector_file)[0]
|
| 106 |
vendor_vectors = pickle.load(f)
|
| 107 |
vendor_data[vendor_name] = vendor_vectors
|
| 108 |
with open(os.path.join('data', score_file), 'rb') as f:
|
|
|
|
| 102 |
vendor_scores = {}
|
| 103 |
for csv_file, vector_file, score_file in zip(csv_files, vector_files, score_files):
|
| 104 |
with open(os.path.join('data', vector_file), 'rb') as f:
|
| 105 |
+
vendor_name = os.path.splitext(vector_file)[0].replace('_vectors','')
|
| 106 |
vendor_vectors = pickle.load(f)
|
| 107 |
vendor_data[vendor_name] = vendor_vectors
|
| 108 |
with open(os.path.join('data', score_file), 'rb') as f:
|