Get desired output

app.py

@@ -52,11 +52,12 @@ def predict(input_image: Image.Image):
             # Read the output from predict.txt
             if os.path.exists('predict.txt'):
                 with open('predict.txt', 'r') as f:
-                    output = f.read().strip().split('\n')[-1]
+                    output_line = f.read().strip().split('\n')[-1]   # Last line
+                    _, prediction = output_line.split(' ', 1)  # split only at the first space
             else:
                 print('predict.txt does not exist')
 
-        return output
+        return prediction
 
     except subprocess.CalledProcessError as e:
         return f"Command failed with error {e.returncode}"