#!/bin/bash

echo "Running setup.sh..."

# Ensure all required Python packages are installed
pip install --upgrade pip
pip install -r requirements.txt

# Manually download the spaCy model
echo "Downloading spaCy model..."
python -m spacy download en_core_web_sm

# Ensure Hugging Face model is downloaded
echo "Downloading Hugging Face model..."
python -c "from transformers import AutoModel, AutoTokenizer; AutoModel.from_pretrained('emilyalsentzer/Bio_ClinicalBERT'); AutoTokenizer.from_pretrained('emilyalsentzer/Bio_ClinicalBERT')"

echo "Setup completed!"