Update app.py

app.py

@@ -47,7 +47,7 @@ def process_model(
     )
     print("Model downloaded successully!")
 
-    metadata_update = f"yes | python3 llama.cpp/gguf-py/scripts/gguf_set_metadata.py {MODEL_NAME}/{FILE_PATH} {key} {value}"
+    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!")
 
@@ -100,7 +100,10 @@ with gr.Blocks() as demo:
                 repo_id=model_id,
                 recursive=True,
             ):
-                files.append(file.path)
+                if "of-0000" in file.path and "0001-of-" not in file.path:
+                    pass
+                elif file.path.endswith("gguf"):
+                    files.append(file.path)
 
             return gr.update(visible=False), gr.update(visible=True, choices=files)
         except Exception: