Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -155,7 +155,7 @@ def evaluate(model, dset_loader, criterion, is_print=False):
|
|
155 |
# 'confidence': confidence
|
156 |
# })
|
157 |
f = open("/home/user/app/result/ho.txt", 'a')
|
158 |
-
f.
|
159 |
|
160 |
if is_print:
|
161 |
print()
|
|
|
155 |
# 'confidence': confidence
|
156 |
# })
|
157 |
f = open("/home/user/app/result/ho.txt", 'a')
|
158 |
+
f.write(prediction+'\n')
|
159 |
|
160 |
if is_print:
|
161 |
print()
|