Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-import whisper
+from faster_whisper import WhisperModel
 from googletrans import Translator
 import tempfile
 
-# Load model (safe)
-model = whisper.load_model("tiny")
+# Load model (CPU safe)
+model = WhisperModel("tiny", device="cpu", compute_type="int8")
 translator = Translator()
 
 def format_time(seconds):
@@ -15,23 +15,22 @@ def format_time(seconds):
     return f"{hrs:02}:{mins:02}:{secs:02},{ms:03}"
 
 def process(video):
-    result = model.transcribe(video)
-    segments = result["segments"]
+    segments, _ = model.transcribe(video)
 
     full_text = ""
     srt_content = ""
 
     for i, seg in enumerate(segments):
-        start = format_time(seg["start"])
-        end = format_time(seg["end"])
-        text = seg["text"].strip()
+        start = format_time(seg.start)
+        end = format_time(seg.end)
+        text = seg.text.strip()
 
         burmese = translator.translate(text, dest="my").text
 
         full_text += burmese + " "
         srt_content += f"{i+1}\n{start} --> {end}\n{burmese}\n\n"
 
-    # Save SRT file
+    # Save SRT
     tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".srt")
     tmp.write(srt_content.encode("utf-8"))
     tmp.close()
@@ -45,7 +44,7 @@ demo = gr.Interface(
         gr.Textbox(label="🇲🇲 Burmese Script"),
         gr.File(label="⬇️ Download SRT")
     ],
-    title="🔥 Burmese Video Subtitle AI (FULL SYSTEM)"
+    title="🔥 Burmese Video AI (FINAL FIX)"
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)