Spaces:
Runtime error
Runtime error
Upload app.py with huggingface_hub
Browse files
app.py
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
import pandas as pd
|
| 3 |
+
|
| 4 |
+
MODELS = [
|
| 5 |
+
{"Model": "SmolLM2-135M", "Speed_tps": 59.7, "Size_MB": 101, "Phone_tps": 46.0},
|
| 6 |
+
{"Model": "TinyLlama-1.1B-int4", "Speed_tps": 17.6, "Size_MB": 400, "Phone_tps": ""},
|
| 7 |
+
{"Model": "Qwen2.5-0.5B-Server", "Speed_tps": 16.7, "Size_MB": 1000, "Phone_tps": ""},
|
| 8 |
+
{"Model": "Qwen2.5-0.5B-Chinese", "Speed_tps": 16.4, "Size_MB": 1000, "Phone_tps": ""},
|
| 9 |
+
{"Model": "Qwen2.5-Math-1.5B", "Speed_tps": 15.7, "Size_MB": 3000, "Phone_tps": ""},
|
| 10 |
+
{"Model": "Qwen2.5-0.5B-int4", "Speed_tps": 12.5, "Size_MB": 469, "Phone_tps": 23.2},
|
| 11 |
+
{"Model": "Llama-3.2-1B-Q4", "Speed_tps": 11.3, "Size_MB": 770, "Phone_tps": 5.4},
|
| 12 |
+
{"Model": "TinyLlama-1.1B-Q5", "Speed_tps": 10.9, "Size_MB": 800, "Phone_tps": ""},
|
| 13 |
+
{"Model": "Gemma-2-2B-IT", "Speed_tps": 8.2, "Size_MB": 5000, "Phone_tps": ""},
|
| 14 |
+
{"Model": "Phi-3.5-mini", "Speed_tps": 8.6, "Size_MB": 7600, "Phone_tps": ""},
|
| 15 |
+
{"Model": "Llama-3.2-3B", "Speed_tps": 4.8, "Size_MB": 6000, "Phone_tps": ""},
|
| 16 |
+
]
|
| 17 |
+
|
| 18 |
+
df = pd.DataFrame(MODELS)
|
| 19 |
+
|
| 20 |
+
with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Size vs Speed") as demo:
|
| 21 |
+
gr.Markdown("# 📈 Model Size vs Speed")
|
| 22 |
+
gr.Markdown("Smaller + faster (top-left) = best for mobile.")
|
| 23 |
+
gr.ScatterPlot(df, x="Size_MB", y="Speed_tps", tooltip="Model",
|
| 24 |
+
title="Size vs Speed (top-left = best for mobile)", height=500)
|
| 25 |
+
gr.Markdown("**Best for mobile:** SmolLM2-135M (101MB, 59.7 t/s)")
|
| 26 |
+
gr.Markdown("---\n🚀 [dispatchAI](https://huggingface.co/dispatchAI)")
|
| 27 |
+
demo.launch()
|