Spaces:
Runtime error
Runtime error
Sebastian Tinoco
commited on
Commit
•
9846887
1
Parent(s):
3a1f89a
changed default model
Browse files- .gitignore +3 -0
- app.py +3 -1
- 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 = '
|
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()
|