Fixes

app.py

@@ -1,25 +1,17 @@
 import sys
 import time
 
-# import torch
 import gradio as gr
 
 from nemo import __version__ as nemo_version
 from nemo.collections.nlp.models import PunctuationCapitalizationModel
 
 # Config
-model_name = "ai-forever/T5-large-spell"
-
+model_name = "dchaplinsky/punctuation_uk_bert"
 concurrency_limit = 5
 
-# Torch
-# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-# torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
-
 # Load the model
-model = PunctuationCapitalizationModel.from_pretrained(
-    "dchaplinsky/punctuation_uk_bert"
-)
+model = PunctuationCapitalizationModel.from_pretrained(model_name)
 
 examples = [
     "тема про яку не люблять говорити офіційні джерела у генштабі і міноборони це хімічна зброя окупанти вже тривалий час використовують хімічну зброю заборонену",