Update app.py

app.py

@@ -1,9 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the Whisper model from Hugging Face
+# Specify the correct model name and adjust the loading configuration
 model_name = "Subhaka/whisper-small-Sinhala-Fine_Tune"
-transcriber = pipeline("automatic-speech-recognition", model=model_name)
+
+# Define the pipeline with `use_safetensors=True`
+def load_pipeline():
+    return pipeline(
+        "automatic-speech-recognition",
+        model=model_name,
+        use_safetensors=True,  # Ensure compatibility with safetensors
+    )
+
+# Load the model pipeline
+transcriber = load_pipeline()
 
 # Define a transcription function
 def transcribe_audio(audio_file):