Finishing touches on final Streamlit version

app.py

@@ -53,11 +53,11 @@ if st.button("RUN!"):
         for i, e in enumerate(sorted(upload_img2, key=lambda x: x.name)):
             with open(f"poses/{i:04}.png", "wb") as f:
                 f.write(e.read())
-        st.info("努力推理中...")
+        st.info("Thinking...")
         os.system("sh infer.sh")
         st.info("Done!")
         video_file=open("output.mp4", "rb")
         video_bytes = video_file.read()
         st.video(video_bytes, start_time=0)
     else:
-        st.info("还没上传图片呢> <")
+        st.info("Please upload a picture")

data_loader.py

@@ -225,7 +225,7 @@ class FileDataset(Dataset):
 
         if "labels" in sample:
             # return UDP as 4chn XYZV float tensor
-            if sample["labels"].dtype != np.float32:
+            if "float" not in str(sample["labels"].dtype):
                 sample["labels"] = sample["labels"].astype(np.float32) / np.iinfo(sample["labels"].dtype).max
             sample["labels"] = torch.from_numpy(
                 sample["labels"].transpose((2, 0, 1)))