Sebastian Tinoco commited on
Commit
9846887
1 Parent(s): 3a1f89a

changed default model

Browse files
Files changed (3) hide show
  1. .gitignore +3 -0
  2. app.py +3 -1
  3. utils.py +1 -1
.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ __pycache__
2
+ *.ipynb
3
+ .env
app.py CHANGED
@@ -4,10 +4,12 @@ import gradio as gr
4
  import os
5
  from huggingface_hub import login
6
  from utils import parse_args
7
-
8
 
9
  args = parse_args()
10
 
 
 
11
  model = args['model']
12
  model_name = "clinical-assistance/" + model
13
 
 
4
  import os
5
  from huggingface_hub import login
6
  from utils import parse_args
7
+ from dotenv import load_dotenv
8
 
9
  args = parse_args()
10
 
11
+ load_dotenv()
12
+
13
  model = args['model']
14
  model_name = "clinical-assistance/" + model
15
 
utils.py CHANGED
@@ -6,7 +6,7 @@ def parse_args():
6
  parser = argparse.ArgumentParser()
7
 
8
  # env arguments
9
- parser.add_argument('--model', type = str, default = 'whisper_medium_baseline_1', help = 'whisper model for inference')
10
 
11
  # consolidate args
12
  args = parser.parse_args()
 
6
  parser = argparse.ArgumentParser()
7
 
8
  # env arguments
9
+ parser.add_argument('--model', type = str, default = 'whisper_medium_clinical_assistance_10k', help = 'whisper model for inference')
10
 
11
  # consolidate args
12
  args = parser.parse_args()