nimool commited on
Commit
f946ebf
·
1 Parent(s): 87448d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -43,12 +43,16 @@ def parse_transcription(logits):
43
 
44
 
45
  # Function to retrieve an answer based on a question (using fuzzy matching)
46
- def get_answer(wav_file):
47
- input_values = read_file_and_process(wav_file)
48
- with torch.no_grad():
49
- logits = model(**input_values).logits
50
- user_question = parse_transcription(logits)
51
-
 
 
 
 
52
  highest_score = 0
53
  best_answer = None
54
 
@@ -68,14 +72,21 @@ model_id = "jonatasgrosman/wav2vec2-large-xlsr-53-persian"
68
  processor = Wav2Vec2Processor.from_pretrained(model_id)
69
  model = Wav2Vec2ForCTC.from_pretrained(model_id)
70
 
71
- input_ = gr.Audio(source="microphone",
 
72
  type="filepath",
73
  label="لطفا دکمه ضبط صدا را بزنید و شروع به صحبت کنید و بعذ از اتمام صحبت دوباره دکمه ضبط را فشار دهید.",
74
  show_download_button=True,
75
  show_edit_button=True,
76
- )
 
 
 
 
 
 
77
  txtbox = gr.Textbox(
78
- label="متن گفتار شما: ",
79
  lines=5,
80
  text_align="right",
81
  show_label=True,
 
43
 
44
 
45
  # Function to retrieve an answer based on a question (using fuzzy matching)
46
+ def get_answer(wav_file, text=None):
47
+
48
+ if text=None:
49
+ input_values = read_file_and_process(wav_file)
50
+ with torch.no_grad():
51
+ logits = model(**input_values).logits
52
+ user_question = parse_transcription(logits)
53
+ elif text != None:
54
+ user_question = text
55
+
56
  highest_score = 0
57
  best_answer = None
58
 
 
72
  processor = Wav2Vec2Processor.from_pretrained(model_id)
73
  model = Wav2Vec2ForCTC.from_pretrained(model_id)
74
 
75
+ input_ = [
76
+ gr.Audio(source="microphone",
77
  type="filepath",
78
  label="لطفا دکمه ضبط صدا را بزنید و شروع به صحبت کنید و بعذ از اتمام صحبت دوباره دکمه ضبط را فشار دهید.",
79
  show_download_button=True,
80
  show_edit_button=True,
81
+ ),
82
+ gr.TextBox(label="سوال خود را بنویسید.",
83
+ lines=3,
84
+ text_align="right",
85
+ show_label=True,)
86
+ ]
87
+
88
  txtbox = gr.Textbox(
89
+ label="پاسخ شما: ",
90
  lines=5,
91
  text_align="right",
92
  show_label=True,