lemms commited on
Commit
665e47c
·
verified ·
1 Parent(s): 8f3458c

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -240,6 +240,13 @@ class RealOpenLLMInference:
240
  "hf_repo": "lemms/openllm-small-extended-9k",
241
  "training_steps": 9000,
242
  "parameters": "35.8M"
 
 
 
 
 
 
 
243
  }
244
  }
245
 
@@ -492,6 +499,7 @@ def create_interface():
492
  | **7k Model** | 7,000 | 35.8M | Enhanced quality (Loss: 2.100, Perplexity: 8.200) |
493
  | **8k Model** | 8,000 | 35.8M | Sophisticated understanding |
494
  | **9k Model** | 9,000 | 35.8M | Best performing model |
 
495
 
496
  **These are real GPT-style transformer models trained on Wikipedia passages from the SQuAD dataset.**
497
 
@@ -503,14 +511,14 @@ def create_interface():
503
  # Model selection
504
  model_dropdown = gr.Dropdown(
505
  choices=list(inference_engine.model_configs.keys()),
506
- value="openllm-small-extended-9k",
507
  label="🎯 Select Model",
508
  info="Choose the real trained model to use"
509
  )
510
 
511
  # Model information display
512
  model_info = gr.Markdown(
513
- value=load_model_info("openllm-small-extended-9k"),
514
  label="📋 Model Information"
515
  )
516
 
@@ -611,6 +619,7 @@ def create_interface():
611
  - [7k Model](https://huggingface.co/lemms/openllm-small-extended-7k)
612
  - [8k Model](https://huggingface.co/lemms/openllm-small-extended-8k)
613
  - [9k Model](https://huggingface.co/lemms/openllm-small-extended-9k)
 
614
  """)
615
 
616
  return interface
 
240
  "hf_repo": "lemms/openllm-small-extended-9k",
241
  "training_steps": 9000,
242
  "parameters": "35.8M"
243
+ },
244
+ "openllm-small-extended-10k": {
245
+ "name": "OpenLLM Small (10k steps)",
246
+ "description": "Real model trained for 10,000 steps - Latest extended training",
247
+ "hf_repo": "lemms/openllm-small-extended-10k",
248
+ "training_steps": 10000,
249
+ "parameters": "35.8M"
250
  }
251
  }
252
 
 
499
  | **7k Model** | 7,000 | 35.8M | Enhanced quality (Loss: 2.100, Perplexity: 8.200) |
500
  | **8k Model** | 8,000 | 35.8M | Sophisticated understanding |
501
  | **9k Model** | 9,000 | 35.8M | Best performing model |
502
+ | **10k Model** | 10,000 | 35.8M | Latest extended training |
503
 
504
  **These are real GPT-style transformer models trained on Wikipedia passages from the SQuAD dataset.**
505
 
 
511
  # Model selection
512
  model_dropdown = gr.Dropdown(
513
  choices=list(inference_engine.model_configs.keys()),
514
+ value="openllm-small-extended-10k",
515
  label="🎯 Select Model",
516
  info="Choose the real trained model to use"
517
  )
518
 
519
  # Model information display
520
  model_info = gr.Markdown(
521
+ value=load_model_info("openllm-small-extended-10k"),
522
  label="📋 Model Information"
523
  )
524
 
 
619
  - [7k Model](https://huggingface.co/lemms/openllm-small-extended-7k)
620
  - [8k Model](https://huggingface.co/lemms/openllm-small-extended-8k)
621
  - [9k Model](https://huggingface.co/lemms/openllm-small-extended-9k)
622
+ - [10k Model](https://huggingface.co/lemms/openllm-small-extended-10k)
623
  """)
624
 
625
  return interface