Update app.py

app.py

@@ -9,6 +9,23 @@ import psutil
 # Define path for HF cache to clean
 HF_CACHE_DIR = os.path.expanduser("~/.cache/huggingface/hub")
 
+# List of models for autocomplete
+MODELS = [
+    'HuggingFaceTB/SmolLM2-135M', 'AxiomicLabs/GPT-X2-125M', 'AxiomicLabs/GPT-X-125M',
+    'facebook/MobileLLM-R1-140M-base', 'SupraLabs/Supra-50M-Base', 'CompactAI-O/Shard-1',
+    'SupraLabs/Supra-50M-Instruct', 'HuggingFaceTB/SmolLM-135M', 'facebook/opt-125m',
+    'AxiomicLabs/GPT-S-5M', 'openai-community/gpt2', 'LH-Tech-AI/Spark-5M-Base-v4',
+    'SupraLabs/Supra-Mini-v5-8M', 'EleutherAI/pythia-70m', 'SupraLabs/Supra-Mini-v4-2M',
+    'EleutherAI/pythia-31m', 'StentorLabs/Stentor3-50M', 'StentorLabs/Stentor3-20M',
+    'StentorLabs/Portimbria-150M', 'HuggingFaceTB/nanowhale-100m-base', 'EleutherAI/pythia-14m',
+    'Harley-ml/Tenete-8M', 'Harley-ml/Dillion-1.2M', 'MihaiPopa-1/CinnabarLM-1.4M-Base',
+    'MihaiPopa-1/CinnabarLM-4M-Base', 'MihaiPopa-1/PotentSulfurLM-500K-Base',
+    'MihaiPopa-1/CinnabarLM-1.5M-Base', 'Harley-ml/Dillionv2-1.3M', 'Eclipse-Senpai/KeyLM-75M',
+    'SupraLabs/Supra-Mini-v6-1M', 'AxiomicLabs/GPT-S-1.4M', 'GODELEV/Archaea-74M',
+    'Sandroeth/cali-0.1B', 'veyra-ai/veyra3-5m-base', 'veyra-ai/veyra-30m-base-5b-tokens',
+    'ThingAI/Quark-50m', 'ThingAI/Quark-135m'
+]
+
 def get_system_stats():
     """Returns a dictionary of current system metrics with formatted strings."""
     mem = psutil.virtual_memory()
@@ -54,9 +71,7 @@ def clean_cache():
     return "Cache directory not found."
 
 # Gradio Interface
-with gr.Blocks(title="Small Model Tester") as app:
-    gr.Markdown("# Advanced HF Model Tester (Free Tier)")
-    
+with gr.Blocks(title="Small MF Model Tester") as app:
 
     current_model = gr.State(None)
     
@@ -69,7 +84,7 @@ with gr.Blocks(title="Small Model Tester") as app:
                 gr.Timer(5).tick(get_system_stats, None, stats_output)
 
             with gr.Row():
-                model_id_input = gr.Textbox(label="Model ID", value="AxiomicLabs/GPT-S-5M", placeholder="e.g., AxiomicLabs/GPT-S-5M", show_label=False)
+                model_id_input = gr.Dropdown(choices=MODELS, label="Model", allow_custom_value=True)
                 load_btn = gr.Button("Load", variant="secondary")
             
             max_tokens_input = gr.Slider(minimum=10, maximum=1024, value=128, step=1, label="Max Output Tokens")