Update app.py

app.py

@@ -29,8 +29,8 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
 # Configuration for model download and conversion
 OLMOASR_REPO = "allenai/OLMoASR"  # Temporary model link as requested
-CHECKPOINT_FILENAME = "OLMoASR-tiny.en.pt"  # Adjust based on actual filename in the repo
-LOCAL_CHECKPOINT_DIR = "checkpoints"
+CHECKPOINT_FILENAME = "tiny.en.pt"  # Adjust based on actual filename in the repo
+LOCAL_CHECKPOINT_DIR = "checkpoint_tiny"
 HF_MODEL_DIR = "tiny_hf"
 
 
@@ -124,7 +124,8 @@ def initialize_models():
     # Download and convert HuggingFace model
     #checkpoint_path = download_olmoasr_checkpoint()
     #hf_model_path = convert_checkpoint_to_hf(checkpoint_path)
-    olmoasr_ckpt = HF_MODEL_DIR
+    olmoasr_ckpt = os.path.join(LOCAL_CHECKPOINT_DIR, CHECKPOINT_FILENAME)
+    hf_model_path = HF_MODEL_DIR
 
     # Load HuggingFace model
     hf_model = AutoModelForSpeechSeq2Seq.from_pretrained(