Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ MODELS = [
|
|
| 25 |
'MihaiPopa-1/CinnabarLM-1.5M-Base', 'Harley-ml/Dillionv2-1.3M', 'Eclipse-Senpai/KeyLM-75M',
|
| 26 |
'SupraLabs/Supra-Mini-v6-1M', 'AxiomicLabs/GPT-S-1.4M', 'GODELEV/Archaea-74M',
|
| 27 |
'Sandroeth/cali-0.1B', 'veyra-ai/veyra3-5m-base', 'veyra-ai/veyra-30m-base-5b-tokens',
|
| 28 |
-
'ThingAI/Quark-50m', 'ThingAI/Quark-135m'
|
| 29 |
]
|
| 30 |
|
| 31 |
# Global class to safely manage the loaded model and tokenizer in memory
|
|
@@ -137,7 +137,7 @@ with gr.Blocks(title="Small MF Model Tester", theme=gr.themes.Soft()) as app:
|
|
| 137 |
gr.Timer(2).tick(get_system_stats, None, stats_output)
|
| 138 |
|
| 139 |
with gr.Group():
|
| 140 |
-
gr.Markdown("##
|
| 141 |
with gr.Row():
|
| 142 |
model_id_input = gr.Dropdown(choices=MODELS, label="Model", allow_custom_value=True, show_label=False, scale=3)
|
| 143 |
load_btn = gr.Button("Load", variant="secondary", scale=1)
|
|
|
|
| 25 |
'MihaiPopa-1/CinnabarLM-1.5M-Base', 'Harley-ml/Dillionv2-1.3M', 'Eclipse-Senpai/KeyLM-75M',
|
| 26 |
'SupraLabs/Supra-Mini-v6-1M', 'AxiomicLabs/GPT-S-1.4M', 'GODELEV/Archaea-74M',
|
| 27 |
'Sandroeth/cali-0.1B', 'veyra-ai/veyra3-5m-base', 'veyra-ai/veyra-30m-base-5b-tokens',
|
| 28 |
+
'ThingAI/Quark-50m', 'ThingAI/Quark-135m', 'HuggingFaceTB/SmolLM2-135M-Instruct'
|
| 29 |
]
|
| 30 |
|
| 31 |
# Global class to safely manage the loaded model and tokenizer in memory
|
|
|
|
| 137 |
gr.Timer(2).tick(get_system_stats, None, stats_output)
|
| 138 |
|
| 139 |
with gr.Group():
|
| 140 |
+
gr.Markdown("## Model Loader\nSelect or Paste custom model id here...")
|
| 141 |
with gr.Row():
|
| 142 |
model_id_input = gr.Dropdown(choices=MODELS, label="Model", allow_custom_value=True, show_label=False, scale=3)
|
| 143 |
load_btn = gr.Button("Load", variant="secondary", scale=1)
|