missed dummy

labdao · Sep 8, 2022 · 97f38f8 · 97f38f8
1 parent 99aa716
commit 97f38f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -93,7 +93,7 @@ def predict(
         ouput_name = os.listdir(args.output_directory + '/dummy')[0]
         print(ouput_name)
 
-        return(args.output_directory + ouput_name)
+        return(args.output_directory + '/dummy/' + ouput_name)
 
 if __name__ == '__main__':
     p = Predictor()