init commut

app.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+
+import torch
+import gradio as gr
+
+from sbert_punc_case_ru import SbertPuncCase
+
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+sbertpunc = SbertPuncCase().to(device)
+
+
+def punctuate(input_string):
+    return sbertpunc.punctuate(input_string)
+
+
+demo = gr.Interface(
+    fn=punctuate,
+    inputs="text",
+    outputs="text",
+    examples=["sbert punc case расставляет точки запятые и знаки вопроса вам нравится"],
+    title="SbertPuncCase",
+    description="SbertPuncCase - модель восстановления пунктуации и регистра для русского языка. Модель способна расставлять точки, запятые и знаки вопроса; определять регистр - слово в нижнем регистре, слово с первой буквой в верхнем регистре, слово в верхнем регистре. Модель разработана для восстановления текста после распознавания речи, поэтому работает со строками в нижнем регистре. В основу модели легла sbert_large_nlu_ru. В качестве обучающих данных использованы текстовые расшифровки интервью.",
+    allow_flagging="never",
+)
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+torch
+gradio
+git+https://huggingface.co/kontur-ai/sbert_punc_case_ru