atereoyinn commited on
Commit
7a56ff8
1 Parent(s): cd3bfde

removed the demo.launch args and chnaged whisper to base

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "large-v3"
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(show_api=False, enable_queue=False, max_timeout=300)
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