Update README.md

README.md

@@ -14,7 +14,7 @@ tags:
 - Prompt Classification
 ---
 
-# 🔁 SimplePromptRouter — классификатор промптов (русский)
+# 🔁 SimpleClassifier — классификатор промптов (русский)
 
 **Кратко:** модель классифицирует входные промпты/вопросы на три действия:
 - **0 — Поиск в локальной базе знаний (RAG)**: сначала ищем релевантные документы в локальном индексе и формируем контекст для генерации.  
@@ -51,7 +51,7 @@ import json, pickle, torch
 import numpy as np
 from types import SimpleNamespace
 
-REPO = "username/SimplePromptRouter"
+REPO = "Neweret/SimplePromptRouter"
 
 config_path = hf_hub_download(REPO, "config.json")
 weights_path = hf_hub_download(REPO, "pytorch_model.bin")
@@ -65,7 +65,7 @@ le_path = hf_hub_download(REPO, "label_encoder.pkl")
 
 cfg = SimpleNamespace(**json.load(open(config_path, "r", encoding="utf-8")))
 
-# --- Динамическая модель (вставь ту же архитектуру, что использовал при обучении) ---
+# --- Динамическая модель ---
 class SimpleClassifier(torch.nn.Module):
     def __init__(self, input_dim, num_classes, p_dropout=0.3):
         super().__init__()
@@ -107,4 +107,4 @@ def predict(text):
     return pred, le.inverse_transform([pred])[0]
 
 # пример
-print(predict("Как мне найти документацию по OpenAI?"))
+print(predict("Как мне найти документацию по нашей компании?"))