Update README.md
Browse files
README.md
CHANGED
|
@@ -9,10 +9,75 @@ sdk_version: 5.33.0
|
|
| 9 |
tags:
|
| 10 |
- mcp-server-track
|
| 11 |
---
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
<video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/6615b9363abccce37d27bbcc/EJvYAY7iO0gsD2hA4NRXZ.mp4"></video>
|
| 14 |
|
| 15 |
-
|
| 16 |
<video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/6615b9363abccce37d27bbcc/bS0aNHLPR8leU2j87awql.qt"></video>
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 9 |
tags:
|
| 10 |
- mcp-server-track
|
| 11 |
---
|
| 12 |
+
A powerful web-based tool for comprehensive technical analysis and visualization of AI models. Upload ONNX models or describe your architecture in text, and get deep insights powered by multiple LLMs (Claude, GPT, Gemini, Mistral) with advanced visualizations (SHAP, LIME, Attention, Architecture, Parameters).
|
| 13 |
+
|
| 14 |
+
## Features
|
| 15 |
+
- **Multi-LLM Analysis:** Choose from Anthropic Claude, OpenAI GPT, Google Gemini, or Mistral for model interpretation.
|
| 16 |
+
- **ONNX & Text Input:** Upload ONNX files or describe your model in plain text/code.
|
| 17 |
+
- **Comprehensive Reports:** Get detailed technical analysis, architecture breakdown, and use-case assessment.
|
| 18 |
+
- **Advanced Visualizations:** Auto-generates SHAP, LIME, attention heatmaps, architecture diagrams, and parameter distributions using Plotly.
|
| 19 |
+
- **Modern Gradio UI:** Intuitive, tabbed interface for easy exploration and comparison.
|
| 20 |
+
|
| 21 |
+
## Installation
|
| 22 |
+
1. Clone this repository:
|
| 23 |
+
```bash
|
| 24 |
+
git clone https://huggingface.co/spaces/Agents-MCP-Hackathon/Model_Analysis_MCP/
|
| 25 |
+
cd Model_Analysis_MCP
|
| 26 |
+
```
|
| 27 |
+
2. Install dependencies:
|
| 28 |
+
```bash
|
| 29 |
+
pip install -r requirements.txt
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
## Configuration
|
| 33 |
+
This app requires API keys for the LLMs you wish to use. Set the following environment variables (e.g., in a `.env` file):
|
| 34 |
+
- `ANTHROPIC_API_KEY` (for Claude)
|
| 35 |
+
- `OPENAI_API_KEY` (for GPT)
|
| 36 |
+
- `GOOGLE_API_KEY` or `GEMINI_API_KEY` (for Gemini)
|
| 37 |
+
- `MISTRAL_API_KEY` (for Mistral)
|
| 38 |
+
|
| 39 |
+
Example `.env`:
|
| 40 |
+
```
|
| 41 |
+
OPENAI_API_KEY=sk-...
|
| 42 |
+
ANTHROPIC_API_KEY=sk-ant-...
|
| 43 |
+
GOOGLE_API_KEY=...
|
| 44 |
+
MISTRAL_API_KEY=...
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
## Usage
|
| 48 |
+
1. Launch the app:
|
| 49 |
+
```bash
|
| 50 |
+
python app.py
|
| 51 |
+
```
|
| 52 |
+
2. Open your browser and go to [http://localhost:7860](http://localhost:7860)
|
| 53 |
+
3. Choose to upload an ONNX model or enter a text description of your model.
|
| 54 |
+
4. Select which LLMs and analysis types you want.
|
| 55 |
+
5. Click **Analyze Model** to generate reports and visualizations.
|
| 56 |
+
|
| 57 |
+
## Example Inputs
|
| 58 |
+
- "ResNet-50 convolutional neural network with 50 layers, batch normalization, and residual connections for image classification"
|
| 59 |
+
- "BERT transformer model with 12 layers, 768 hidden dimensions, and multi-head attention for NLP tasks"
|
| 60 |
+
- "LSTM recurrent neural network with 256 hidden units for time series prediction"
|
| 61 |
+
- "U-Net architecture with encoder-decoder structure and skip connections for image segmentation"
|
| 62 |
+
|
| 63 |
+
## Demo
|
| 64 |
+
|
| 65 |
+
### Cursor:
|
| 66 |
<video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/6615b9363abccce37d27bbcc/EJvYAY7iO0gsD2hA4NRXZ.mp4"></video>
|
| 67 |
|
| 68 |
+
### Gradio UI:
|
| 69 |
<video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/6615b9363abccce37d27bbcc/bS0aNHLPR8leU2j87awql.qt"></video>
|
| 70 |
|
| 71 |
+
## Requirements
|
| 72 |
+
See `requirements.txt` for the full list. Key packages:
|
| 73 |
+
- gradio
|
| 74 |
+
- openai, anthropic, mistralai, google-generativeai
|
| 75 |
+
- onnx, onnxruntime
|
| 76 |
+
- numpy, pandas, plotly, networkx
|
| 77 |
+
- python-dotenv
|
| 78 |
+
|
| 79 |
+
## References
|
| 80 |
+
- [Gradio Documentation](https://gradio.app/docs/)
|
| 81 |
+
- [ONNX](https://onnx.ai/)
|
| 82 |
+
|
| 83 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|