RootLinux21 commited on
Commit
af9ce38
·
verified ·
1 Parent(s): 6817bc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ if torch.cuda.is_available():
25
  print(f"GPU: {torch.cuda.get_device_name(0)}")
26
 
27
  # СВОЯ НЕЙРОСЕТЬ
28
- MODEL_NAME = "DeepPavlov/rubert-base-cased" # Для токенизации
29
  EMBEDDING_MODEL = "all-MiniLM-L6-v2"
30
  SCIENCE_DATASET = "RafaelUI/ru_science"
31
  ARTICLE_LIMIT = 200 # Для обучения
 
25
  print(f"GPU: {torch.cuda.get_device_name(0)}")
26
 
27
  # СВОЯ НЕЙРОСЕТЬ
28
+ MODEL_NAME = "Qwen/Qwen-1.5B-1.8B-Chat" # Для токенизации
29
  EMBEDDING_MODEL = "all-MiniLM-L6-v2"
30
  SCIENCE_DATASET = "RafaelUI/ru_science"
31
  ARTICLE_LIMIT = 200 # Для обучения