fix script

Dockerfile.hardware

@@ -73,8 +73,9 @@ RUN apt update && apt install -y \
 COPY --from=pkg-builder /ros_ws /ros_ws
 
 # copy firmware built in previous stage
-RUN curl -LJO https://github.com/husarion/rosbot-stm32-firmware/releases/download/$ROSBOT_FW_RELEASE/firmware_diff.bin -o /root/firmware_diff.bin && \
-	curl -LJO https://github.com/husarion/rosbot-stm32-firmware/releases/download/$ROSBOT_FW_RELEASE/firmware_bin.bin -o /root/firmware_mecanum.bin
+RUN cd root && \
+	curl https://github.com/husarion/rosbot-stm32-firmware/releases/download/$ROSBOT_FW_RELEASE/firmware_diff.bin -o firmware_diff.bin && \
+	curl https://github.com/husarion/rosbot-stm32-firmware/releases/download/$ROSBOT_FW_RELEASE/firmware_mec.bin -o firmware_mecanum.bin
 
 # copy scripts
 COPY ./flash-firmware.py /

flash-firmware.py

@@ -30,21 +30,21 @@ def __init__(self, sys_arch, binary_file):
 
         print(f"System architecture: {self.sys_arch}")
 
-        if self.sys_arch.stdout == b"armv7l\n":
+        if self.sys_arch == "armv7l\n":
             # Setups ThinkerBoard pins
             print("Device: ThinkerBoard\n")
             self.port = "/dev/ttyS1"
             boot0_pin_no = 164
             reset_pin_no = 184
 
-        elif self.sys_arch.stdout == b"x86_64\n":
+        elif self.sys_arch == "x86_64\n":
             # Setups UpBoard pins
             print("Device: UpBoard\n")
             self.port = "/dev/ttyS4"
             boot0_pin_no = 17
             reset_pin_no = 18
 
-        elif self.sys_arch.stdout == b"aarch64\n":
+        elif self.sys_arch == "aarch64\n":
             # Setups RPi pins
             print("Device: RPi\n")
             self.port = "/dev/ttyAMA0"