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()
|