Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForTextToWaveform
+import torch  # Add the import statement for torch
 
 # Load your pretrained models
 asr_model = Wav2Vec2ForCTC.from_pretrained("Baghdad99/saad-speech-recognition-hausa-audio-to-text")
@@ -24,7 +25,7 @@ def translate_speech(speech):
     # Transcribe the speech to text
     inputs = asr_processor(audio_signal, return_tensors="pt", padding=True)
     logits = asr_model(inputs.input_values).logits
-    predicted_ids = torch.argmax(logits, dim=-1)
+    predicted_ids = torch.argmax(logits, dim=-1)  # Add torch module to access argmax function
     transcription = asr_processor.decode(predicted_ids[0])
 
     # Translate the text
@@ -41,4 +42,3 @@ def translate_speech(speech):
 # Define the Gradio interface
 iface = gr.Interface(fn=translate_speech, inputs=gr.inputs.Audio(source="microphone"), outputs="audio")
 iface.launch()
-