updated to medium model

main.py

@@ -38,11 +38,7 @@
 
 
 # # #uvicorn app:app --host 0.0.0.0 --port 8000 --reload
-# # #curl -X GET "http://localhost:8000/?text=I%20like%20Apples"
-# # #http://localhost:8000/?text=I%20like%20Apples
-
-
-
+#
 
 
 
@@ -144,7 +140,6 @@
 
 
 
-
 # #uvicorn app:app --host 0.0.0.0 --port 8000 --reload
 
 
@@ -213,7 +208,6 @@ import librosa
 import io
 import re
 
-
 html_tag_remover = re.compile(r'<[^>]+>')
 
 def remove_tags(text):
@@ -221,8 +215,8 @@ def remove_tags(text):
 
 app = FastAPI()
 
-processor = WhisperProcessor.from_pretrained("openai/whisper-small")
-model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
+processor = WhisperProcessor.from_pretrained("openai/whisper-medium")
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-medium")
 model.config.forced_decoder_ids = None
 
 chunk_duration = 30 
@@ -267,6 +261,5 @@ async def transcribe_audio(audio_file: UploadFile):
         final_transcription = " ".join(transcription)
         final_transcription = remove_tags(final_transcription)
 
-        
         return final_transcription