clean

app.py

@@ -13,7 +13,6 @@ import gradio.themes as gr_themes
 import numpy as np
 import spaces
 import torch
-from deepmultilingualpunctuation import PunctuationModel
 from huggingface_hub import CommitScheduler, hf_hub_download
 from nemo.collections.asr.models import ASRModel
 from pydub import AudioSegment
@@ -75,13 +74,6 @@ whisper_tiny_model = AutoModelForSpeechSeq2Seq.from_pretrained(
 )
 whisper_tiny_processor = AutoProcessor.from_pretrained(whisper_tiny_model_id)
 
-# Load punctuation restoration model
-print("Loading punctuation restoration model...")
-punctuation_model = PunctuationModel(
-    model="oliverguhr/fullstop-punctuation-multilingual-sonar-base"
-)
-print("Punctuation model loaded successfully.")
-
 
 def start_session(request: gr.Request):
     session_hash = request.session_hash
@@ -172,20 +164,16 @@ def generate_srt_content(
 
 def restore_punctuation(text: str) -> str:
     """
-    Restore punctuation in a text using the punctuation model.
-    Also fixes casing after sentence-ending punctuation (. ? !).
-    Returns the text with punctuation and proper casing.
+    Apply basic text formatting (capitalize first letter).
+    Returns the text with basic formatting.
     """
     if not text or text.strip() == "":
         return text
-    try:
-        punctuated_text = punctuation_model.restore_punctuation(text)
-        # Fix casing after sentence-ending punctuation
-        punctuated_text = fix_sentence_casing(punctuated_text)
-        return punctuated_text
-    except Exception as e:
-        print(f"Warning: Punctuation restoration failed: {e}")
-        return text
+    # Just capitalize the first letter if it's lowercase
+    text = text.strip()
+    if text and text[0].islower():
+        text = text[0].upper() + text[1:]
+    return text
 
 
 def fix_sentence_casing(text: str) -> str:

requirements.txt

@@ -1,5 +1,4 @@
 Cython
 cuda-python
 nemo_toolkit[asr]==2.5.0
-transformers
-deepmultilingualpunctuation
+transformers