3morixd commited on
Commit
4574cbc
·
verified ·
1 Parent(s): b4cc22d

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +27 -0
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()