Update app.py

app.py

@@ -128,7 +128,7 @@ def transcribe_audio_streaming(audio_file, model_name, progress=gr.Progress()):
             # Create prompt tokens for medical context
             prompt = """ Endocrinology clinic visit transcript.
 Previous discussion mentioned levothyroxine dose adjustment for Hashimoto’s thyroiditis and an Hemoglobin A1C review.
-Terms we frequently use: levothyroxine, Synthroid, Cytomel, methimazole, propylthiouracil, PTU semaglutide, Ozempic, Mounjaro, insulin glargine, Lantus, Tresiba, TOujoe, metformin, Jardiance, hydrocortisone, dexamethasone.
+Terms we frequently use: levothyroxine, Synthroid, Cytomel, methimazole, propylthiouracil, PTU semaglutide, Ozempic, Mounjaro, insulin glargine, Lantus, Tresiba, Toujeo, Metformin, Glipizide, Januvia, Jardiance, Atorvastatin, Rosuvastatin, Pravastatin.
 Common labs and hormones: TSH, free T4, free T3, HbA1C, cortisol, ACTH, parathyroid hormone, LDL, microalbumin.
 Typical procedures: thyroid ultrasound, fine-needle aspiration biopsy, FNA, medtronics pump, Tandem pump, OmniPod pump, radioactive iodine ablation, continuous glucose monitoring, CGM, DEXA scan."""
             prompt_ids = loaded_whisper_processor.get_prompt_ids(prompt, return_tensors="pt").to(device)
@@ -441,13 +441,13 @@ MEDICAL_LLM_MODELS = [
 # Create Gradio interface
 with gr.Blocks(title="Medical Transcription & Note Generation", theme=gr.themes.Soft()) as demo:
     gr.HTML("<h1 style='text-align: center; color: #2563eb;'>🏥 Medical Transcription & Note Generation</h1>")
-    gr.HTML("<p style='text-align: center;'>Record patient-doctor conversations, transcribe with Whisper, and generate structured medical notes</p>")
+    gr.HTML("<p style='text-align: center;'>Only for research purpose, not for clinical use.</p>")
     
     # Add warning about HF token
-    if not HF_TOKEN:
-        gr.HTML("<div style='background-color: #fef3cd; border: 1px solid #ffeaa7; padding: 10px; margin: 10px 0; border-radius: 5px;'><strong>⚠️ Warning:</strong> HF_TOKEN not found. Some models may not be accessible. Please add your Hugging Face token as a Space secret.</div>")
-    else:
-        gr.HTML("<div style='background-color: #d4edda; border: 1px solid #c3e6cb; padding: 10px; margin: 10px 0; border-radius: 5px;'><strong>✅ Status:</strong> Hugging Face token loaded successfully.</div>")
+    #if not HF_TOKEN:
+    #    gr.HTML("<div style='background-color: #fef3cd; border: 1px solid #ffeaa7; padding: 10px; margin: 10px 0; border-radius: 5px;'><strong>⚠️ Warning:</strong> HF_TOKEN not found. Some models may not be accessible. Please add your Hugging Face token as a Space secret.</div>")
+    #else:
+    #    gr.HTML("<div style='background-color: #d4edda; border: 1px solid #c3e6cb; padding: 10px; margin: 10px 0; border-radius: 5px;'><strong>✅ Status:</strong> Hugging Face token loaded successfully.</div>")
     
     with gr.Row():
         # Left Column - Audio Recording and Transcription
@@ -464,7 +464,7 @@ with gr.Blocks(title="Medical Transcription & Note Generation", theme=gr.themes.
             # Whisper model selection
             whisper_model = gr.Dropdown(
                 choices=WHISPER_MODELS,
-                value="openai/whisper-small",
+                value="openai/whisper-tiny",
                 label="Select Whisper Model"
             )