Update app.py

app.py

@@ -31,7 +31,7 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 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"
-HF_MODEL_DIR = "checkpoints/tiny_hf_converted"
+HF_MODEL_DIR = "tiny_hf"
 
 
 def ensure_checkpoint_dir():
@@ -122,9 +122,9 @@ def convert_checkpoint_to_hf(checkpoint_path):
 def initialize_models():
     """Initialize both HuggingFace and OLMoASR models."""
     # Download and convert HuggingFace model
-    checkpoint_path = download_olmoasr_checkpoint()
-    hf_model_path = convert_checkpoint_to_hf(checkpoint_path)
-    olmoasr_ckpt = checkpoint_path
+    #checkpoint_path = download_olmoasr_checkpoint()
+    #hf_model_path = convert_checkpoint_to_hf(checkpoint_path)
+    olmoasr_ckpt = HF_MODEL_DIR
 
     # Load HuggingFace model
     hf_model = AutoModelForSpeechSeq2Seq.from_pretrained(