Add application file

app.py

@@ -0,0 +1,48 @@
+import torch
+import gradio as gr
+from transformers import AutoProcessor, pipeline
+from datasets import load_dataset
+
+def load_fama(model_id):
+    processor = AutoProcessor.from_pretrained(model_id)
+
+    device = "cuda:0" if torch.cuda.is_available() else "cpu"
+    tgt_lang = "en"
+
+    # Force the model to start with the language tag
+    lang_tag = "<lang:{}>".format(tgt_lang)
+    lang_tag_id = processor.tokenizer.convert_tokens_to_ids(lang_tag)
+
+    generate_kwargs = {"num_beams": 5, "no_repeat_ngram_size": 5, "forced_bos_token_id": lang_tag_id}
+
+    pipe = pipeline(
+        "automatic-speech-recognition",
+        model=model_id,
+        trust_remote_code=True,
+        torch_dtype=torch.float32,
+        device=device,
+        return_timestamps=False,
+        generate_kwargs=generate_kwargs
+    )
+    return pipe
+
+#load fama model
+model_id = "FBK-MT/fama-small"
+pipeline = load_fama(model_id)
+
+
+def transcribe(audio):
+    """
+    Function called by gradio interface. It runs model inference on an audio sample
+    """
+    dataset = load_dataset("distil-whisper/librispeech_asr_dummy", "clean", split="validation")
+    sample = dataset[0]["audio"]
+    result = pipeline(sample)
+    return result["text"]
+
+
+if __name__ == "__main__":
+    
+    #run gradio interface
+    iface = gr.Interface(fn=transcribe, inputs=gr.Audio(type="filepath"), outputs="text")
+    iface.launch()

requirements.txt

@@ -0,0 +1,111 @@
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.12.7
+aiosignal==1.3.2
+annotated-types==0.7.0
+anyio==4.9.0
+attrs==25.3.0
+audioread==3.0.1
+certifi==2025.4.26
+cffi==1.17.1
+charset-normalizer==3.4.2
+click==8.2.1
+datasets==3.6.0
+decorator==5.2.1
+dill==0.3.8
+fastapi==0.115.12
+ffmpy==0.6.0
+filelock==3.18.0
+frozenlist==1.6.0
+fsspec==2025.3.0
+gradio==5.32.1
+gradio_client==1.10.2
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.1.2
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.32.4
+idna==3.10
+Jinja2==3.1.6
+joblib==1.5.1
+lazy_loader==0.4
+librosa==0.11.0
+llvmlite==0.44.0
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mpmath==1.3.0
+msgpack==1.1.0
+multidict==6.4.4
+multiprocess==0.70.16
+networkx==3.5
+numba==0.61.2
+numpy==2.2.6
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cuda-cupti-cu12==12.6.80
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-cuda-runtime-cu12==12.6.77
+nvidia-cudnn-cu12==9.5.1.17
+nvidia-cufft-cu12==11.3.0.4
+nvidia-cufile-cu12==1.11.1.6
+nvidia-curand-cu12==10.3.7.77
+nvidia-cusolver-cu12==11.7.1.2
+nvidia-cusparse-cu12==12.5.4.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-nccl-cu12==2.26.2
+nvidia-nvjitlink-cu12==12.6.85
+nvidia-nvtx-cu12==12.6.77
+orjson==3.10.18
+packaging==25.0
+pandas==2.2.3
+pillow==11.2.1
+platformdirs==4.3.8
+pooch==1.8.2
+propcache==0.3.1
+pyarrow==20.0.0
+pycparser==2.22
+pydantic==2.11.5
+pydantic_core==2.33.2
+pydub==0.25.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==14.0.0
+ruff==0.11.12
+safehttpx==0.1.6
+safetensors==0.5.3
+scikit-learn==1.6.1
+scipy==1.15.3
+semantic-version==2.10.0
+sentencepiece==0.2.0
+setuptools==80.9.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+soundfile==0.13.1
+soxr==0.5.0.post1
+starlette==0.46.2
+sympy==1.14.0
+threadpoolctl==3.6.0
+tokenizers==0.21.1
+tomlkit==0.13.2
+torch==2.7.0
+torchaudio==2.7.0
+torchvision==0.22.0
+tqdm==4.67.1
+transformers==4.48.1
+triton==3.3.0
+typer==0.16.0
+typing-inspection==0.4.1
+typing_extensions==4.14.0
+tzdata==2025.2
+urllib3==2.4.0
+uvicorn==0.34.3
+websockets==15.0.1
+xxhash==3.5.0
+yarl==1.20.0