import streamlit as st
from transformers import pipeline

# Load your model
model_name = "yash072/Whisper_Smal_FineTuned_Hindi"
pipe = pipeline("automatic-speech-recognition", model=model_name)

def transcribe(audio_file):
    result = pipe(audio_file)
    return result["text"]

st.title("Whisper Small Fine-Tuned for Hindi")

audio_file = st.file_uploader("Upload an audio file", type=["wav", "mp3", "ogg"])

if audio_file is not None:
    transcription = transcribe(audio_file)
    st.write("Transcription:")
    st.write(transcription)