rty to fix errors

app.py

@@ -14,7 +14,7 @@ def transcribe_audio(audio):
     if audio is None:
         return "Please upload an audio file."
     if transcribe_model is None:
-        return "Please select a model."
+        return "Please load the transcription model first."
 
     device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
@@ -41,6 +41,8 @@ def proofread(text):
     global proofread_model
     if text is None:
         return "Please provide the transcribed text for proofreading."
+    if proofread_model is None:
+        return "Please load the proofreading model first."
     
     device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
@@ -77,7 +79,7 @@ def load_models(transcribe_model_id, proofread_model_id):
 with gr.Blocks() as demo:
     gr.Markdown("""
                 # Audio Transcription and Proofreading
-                1. Select models for transcription and proofreading
+                1. Select models for transcription and proofreading and load them
                 2. Upload an audio file (Wait for the file to be fully loaded first)
                 3. Transcribe the audio
                 4. Proofread the transcribed text