Update app_flash1.py

app_flash1.py

@@ -63,7 +63,7 @@ def build_encoder(model_name="gpt2", max_length=128):
 def push_flashpack_model_to_hf(model, hf_repo):
     with tempfile.TemporaryDirectory() as tmp_dir:
         repo = Repository(local_dir=tmp_dir, clone_from=hf_repo, use_auth_token=True)
-        model.save_flashpack(os.path.join(tmp_dir, "model.flashpack"))
+        model.save_flashpack(os.path.join(tmp_dir, "model.flashpack"),target_dtype=torch.float32)
         with open(os.path.join(tmp_dir, "README.md"), "w") as f:
             f.write("# FlashPack Model\nTrained locally and pushed to HF.")
         repo.push_to_hub()