Spaces:
Running
Running
| import gradio as gr | |
| import requests | |
| import json | |
| def fetch_manifest(model_id, tag_name): | |
| try: | |
| model_id = model_id.replace("hf.co", "").replace("://", "").replace("/", " ").strip().replace(" ", "/") | |
| # Fetch manifest | |
| manifest_url = f"https://huggingface.co/v2/{model_id}/manifests/{tag_name}" | |
| manifest_response = requests.get(manifest_url) | |
| manifest_response.raise_for_status() | |
| manifest_data = manifest_response.json() | |
| # Initialize output | |
| output = f"Manifest for {model_id}:{tag_name}\n" | |
| output += json.dumps(manifest_data, indent=2) + "\n" | |
| # Find template and params layers | |
| system_layer = next((layer for layer in manifest_data["layers"] | |
| if layer["mediaType"] == "application/vnd.ollama.image.system"), None) | |
| template_layer = next((layer for layer in manifest_data["layers"] | |
| if layer["mediaType"] == "application/vnd.ollama.image.template"), None) | |
| params_layer = next((layer for layer in manifest_data["layers"] | |
| if layer["mediaType"] == "application/vnd.ollama.image.params"), None) | |
| # Fetch and display system if found | |
| if system_layer: | |
| system_url = f"https://huggingface.co/v2/{model_id}/blobs/{system_layer['digest']}" | |
| system_response = requests.get(system_url) | |
| system_response.raise_for_status() | |
| output += "\n\n======================\n\nSystem message:\n" | |
| output += system_response.text + "\n" | |
| # Fetch and display template if found | |
| if template_layer: | |
| template_url = f"https://huggingface.co/v2/{model_id}/blobs/{template_layer['digest']}" | |
| template_response = requests.get(template_url) | |
| template_response.raise_for_status() | |
| output += "\n\n======================\n\nTemplate:\n" | |
| output += template_response.text + "\n" | |
| # Fetch and display params if found | |
| if params_layer: | |
| params_url = f"https://huggingface.co/v2/{model_id}/blobs/{params_layer['digest']}" | |
| params_response = requests.get(params_url) | |
| params_response.raise_for_status() | |
| output += "\n\n======================\n\nParameters:\n" | |
| output += json.dumps(params_response.json(), indent=2) + "\n" | |
| return output | |
| except requests.exceptions.RequestException as e: | |
| raise gr.Error(f"Error occurred: {str(e)}") | |
| except Exception as e: | |
| raise gr.Error(f"Unexpected error: {str(e)}") | |
| # Create Gradio interface | |
| iface = gr.Interface( | |
| fn=fetch_manifest, | |
| inputs=[ | |
| gr.Textbox(label="Model ID (e.g., bartowski/Meta-Llama-3.1-8B-Instruct-GGUF)", placeholder="Enter model ID"), | |
| gr.Textbox("latest", label="Tag Name (e.g., latest)", placeholder="Enter tag name") | |
| ], | |
| outputs=gr.Code(label="Results", lines=40), | |
| title="Hugging Face Model Manifest Viewer", | |
| description="Enter a Hugging Face model ID and tag name to view its manifest, template, and parameters.", | |
| ) | |
| iface.launch() |