Update app.py

app.py

@@ -27,8 +27,6 @@ def process_model(
     print(f"key: {key}")
     print(f"value: {value}")
 
-    api = HfApi(token=oauth_token.token)
-
     MODEL_NAME = model_id.split("/")[-1]
 
     if file_path_dropdown:
@@ -38,28 +36,33 @@ def process_model(
 
     print(f"FILE_PATH: {FILE_PATH}")
 
-    FILE_NAME = file_path.split("/")[-1]
+    try:
+        api = HfApi(token=oauth_token.token)
 
-    api.snapshot_download(
-        repo_id=model_id,
-        allow_patterns=FILE_PATH,
-        local_dir=f"{MODEL_NAME}",
-    )
-    print("Model downloaded successully!")
+        FILE_NAME = file_path.split("/")[-1]
 
-    metadata_update = f"yes YES | python3 llama.cpp/gguf-py/scripts/gguf_set_metadata.py {MODEL_NAME}/{FILE_PATH} {key} {value}"
-    subprocess.run(metadata_update, shell=True)
-    print(f"Model metadata {key} updated to {value} successully!")
+        api.snapshot_download(
+            repo_id=model_id,
+            allow_patterns=FILE_PATH,
+            local_dir=f"{MODEL_NAME}",
+        )
+        print("Model downloaded successully!")
 
-    # Upload gguf files
-    api.upload_folder(
-        folder_path=MODEL_NAME,
-        repo_id=model_id,
-        allow_patterns=["*.gguf"],
-    )
-    print("Uploaded successfully!")
+        metadata_update = f"python3 llama.cpp/gguf-py/scripts/gguf_set_metadata.py {MODEL_NAME}/{FILE_PATH} {key} {value} --force"
+        subprocess.run(metadata_update, shell=True)
+        print(f"Model metadata {key} updated to {value} successully!")
+
+        # Upload gguf files
+        api.upload_folder(
+            folder_path=MODEL_NAME,
+            repo_id=model_id,
+            allow_patterns=["*.gguf"],
+        )
+        print("Uploaded successfully!")
 
-    return "Processing complete."
+        return f"Processing complete"
+    except Exception as e:
+        return f"Something went wrong: {e}"
 
 
 with gr.Blocks() as demo:
@@ -85,7 +88,6 @@ with gr.Blocks() as demo:
         inputs=[model_id, file_path, file_path_dropdown, key, value],
         outputs=[
             gr.Markdown(label="output"),
-            gr.Image(show_label=False),
         ],
         title="Update metadata for a GGUF file",
         description="The space takes an HF repo, a file within that repo, a metadata key, and new metadata value to update it to.",