added torch no grad

app.py

@@ -32,7 +32,8 @@ def predict_and_ctc_lm_decode(input_file, model_name):
   speech = load_and_fix_data(input_file)
 
   input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
-  logits = model(input_values).logits.cpu().detach().numpy()[0]
+  with torch.no_grad():
+    logits = model(input_values).logits.cpu().detach().numpy()[0]
 
   pred = processor.decode(logits).text
 
@@ -45,7 +46,8 @@ def predict_and_greedy_decode(input_file, model_name):
   speech = load_and_fix_data(input_file)
 
   input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
-  logits = model(input_values).logits
+  with torch.no_grad():
+    logits = model(input_values).logits
 
   predicted_ids = torch.argmax(logits, dim=-1)
   pred = processor.batch_decode(predicted_ids)
@@ -59,11 +61,11 @@ def return_all_predictions(input_file, model_name):
 
 
 gr.Interface(return_all_predictions,
-             inputs = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown(["jonatasgrosman/wav2vec2-large-xlsr-53-spanish", "jonatasgrosman/wav2vec2-xls-r-1b-spanish"], label="Model Name")],
+             inputs = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown(["LuisG07/wav2vec2-large-xlsr-53-spanish", "jonatasgrosman/wav2vec2-xls-r-1b-spanish"], label="Model Name")],
              outputs = [gr.outputs.Textbox(label="Beam CTC decoding w/ LM"), gr.outputs.Textbox(label="Greedy decoding")],
              title="ASR using Wav2Vec2 & pyctcdecode in spanish",
              description = "Comparing greedy decoder with beam search CTC decoder, record/ drop your audio!",
              layout = "horizontal",
-             examples = [["test1.wav", "jonatasgrosman/wav2vec2-large-xlsr-53-spanish"], ["test2.wav", "jonatasgrosman/wav2vec2-xls-r-1b-spanish"]], 
+             examples = [["test1.wav", "LuisG07/wav2vec2-large-xlsr-53-spanish"], ["test2.wav", "jonatasgrosman/wav2vec2-xls-r-1b-spanish"]], 
              theme="huggingface",
              enable_queue=True).launch()