Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import subprocess
 import os
-
+import shutil
 from huggingface_hub import HfApi, snapshot_download
 from gradio_huggingfacehub_search import HuggingfaceHubSearch
 
@@ -41,6 +41,8 @@ def process_model(
 
         FILE_NAME = file_path.split("/")[-1]
 
+        print("Starting download")
+
         api.snapshot_download(
             repo_id=model_id,
             allow_patterns=FILE_PATH,
@@ -52,13 +54,16 @@ def process_model(
         subprocess.run(metadata_update, shell=True)
         print(f"Model metadata {key} updated to {value} successully!")
 
+        print(f"Reuploading file")
         # Upload gguf files
         api.upload_folder(
             folder_path=MODEL_NAME,
+            commit_message=f"Updating {FILE_NAME} metadata {key} to {value}",
             repo_id=model_id,
             allow_patterns=["*.gguf"],
         )
         print("Uploaded successfully!")
+        shutil.rmtree(f"{MODEL_NAME}/")
 
         return f"Processing complete"
     except Exception as e: