Update app.py

app.py

@@ -5,9 +5,11 @@ import gradio as gr
 import sox
 import subprocess
 from fuzzywuzzy import fuzz
-from data import dataset
+from data import get_data
 
 
+dataset = get_data()
+
 def read_file_and_process(wav_file):
     filename = wav_file.split('.')[0]
     filename_16k = filename + "16k.wav"
@@ -41,7 +43,12 @@ def parse(wav_file):
 
 
 # Function to retrieve an answer based on a question (using fuzzy matching)
-def get_answer(user_question):
+def get_answer(wav_file):
+    input_values = read_file_and_process(wav_file)
+     with torch.no_grad():
+        logits = model(**input_values).logits
+    user_question = parse_transcription(logits)
+
     highest_score = 0
     best_answer = None