Update app.py

app.py

@@ -218,7 +218,7 @@ def download_ckpt(ckpt_url_2, ckpt_name_2):
             return "error download ckpt!"
 
 
-def to_safetensors(safetensors_name_2):
+def to_safetensors(ckpt_name_2, safetensors_name_2):
     os.system("mkdir safetensors")
     weights = torch.load(f"ckpt/{ckpt_name_2}.ckpt")
     if "state_dict" in weights:
@@ -278,7 +278,7 @@ def download_safetensors(safetensors_url, safetensors_name):
         except Exception as e:
             return "error download safetensors!"
 
-def from_safetensors_to_ckpt(ckpt_name):
+def from_safetensors_to_ckpt(safetensors_name, ckpt_name):
     weights = load_file(f"safetensors/{safetensors_name}.safetensors", device="cpu")
     os.system("mkdir ckpt")
     torch.save(weights, f"ckpt/{ckpt_name}.ckpt")