atereoyinn
commited on
Commit
•
7a56ff8
1
Parent(s):
cd3bfde
removed the demo.launch args and chnaged whisper to base
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ os.environ["COMMANDLINE_ARGS"] = "--no-gradio-queue"
|
|
17 |
# Initiate checkpoints for model loading
|
18 |
numind_checkpoint = "numind/NuExtract-tiny"
|
19 |
llama_checkpoint = "Atereoyin/Llama3_finetuned_for_medical_entity_extraction"
|
20 |
-
whisper_checkpoint = "
|
21 |
|
22 |
quantization_config = BitsAndBytesConfig(
|
23 |
load_in_8bit=True,
|
@@ -209,6 +209,6 @@ with demo:
|
|
209 |
outputs=gr.File(label="Download CSV")
|
210 |
)
|
211 |
|
212 |
-
demo.launch(
|
213 |
|
214 |
|
|
|
17 |
# Initiate checkpoints for model loading
|
18 |
numind_checkpoint = "numind/NuExtract-tiny"
|
19 |
llama_checkpoint = "Atereoyin/Llama3_finetuned_for_medical_entity_extraction"
|
20 |
+
whisper_checkpoint = "base"
|
21 |
|
22 |
quantization_config = BitsAndBytesConfig(
|
23 |
load_in_8bit=True,
|
|
|
209 |
outputs=gr.File(label="Download CSV")
|
210 |
)
|
211 |
|
212 |
+
demo.launch(max_timeout=300)
|
213 |
|
214 |
|