Update main.py

main.py

@@ -154,6 +154,8 @@ def evaluate(model, dset_loader, criterion, is_print=False):
             #     'prediction': prediction,
             #     'confidence': confidence
             # })
+            with open("/home/user/app/result/ho.txt", 'w') as f:
+                f.writelines(prediction)
 
             if is_print:
                 print()