Upload app.py with huggingface_hub
Browse files
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-
|
| 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-
|
| 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
|