Update Gpt4oDemo.py

Gpt4oDemo.py

@@ -34,18 +34,25 @@ MODEL="gpt-4o"
 device = "cpu" 
 batch_size = 16 # reduce if low on GPU mem
 compute_type = "int8" # change to "int8" if low on GPU mem (may reduce accuracy)
+
+# Import the necessary components for TranscriptionOptions
+from faster_whisper.transcribe import TranscriptionOptions
+
+# Define the required arguments for TranscriptionOptions
 max_new_tokens = 512  # Example value, adjust as needed
 clip_timestamps = True  # Example value, adjust as needed
 hallucination_silence_threshold = 0.5  # Example value, adjust as needed
 
-# Load the model with the required arguments
-default_asr_options = {
-    "max_new_tokens": max_new_tokens,
-    "clip_timestamps": clip_timestamps,
-    "hallucination_silence_threshold": hallucination_silence_threshold
-}
+# Initialize TranscriptionOptions with the required arguments
+default_asr_options = TranscriptionOptions(
+    max_new_tokens=max_new_tokens,
+    clip_timestamps=clip_timestamps,
+    hallucination_silence_threshold=hallucination_silence_threshold
+)
+
+# Load the model using whisperx.load_model
+model = whisperx.load_model("large-v2", device, compute_type=compute_type)
 
-model = whisperx.load_model("large-v2", device, compute_type=compute_type, **default_asr_options)
 '''
 Video
 '''