Spaces:

HumbleBeeAI
/

al-ghazali-rag-retrieval-api

Running

al-ghazali-rag-retrieval-api / download_models.py

Update .gitignore to include .env, add model download script, enhance save_data endpoint for batch processing, and modify model_type formatting in index.html

4fd6d11 12 days ago

raw

history blame contribute delete

230 Bytes

	from transformers import AutoModel, AutoTokenizer

	model_names = [
	"WhereIsAI/UAE-Large-V1",
	"BAAI/bge-large-en-v1.5",
	]

	for name in model_names:
	AutoModel.from_pretrained(name)
	AutoTokenizer.from_pretrained(name)