ContentAgent / agents /model.py
yetessam's picture
Update agents/model.py
f8c087a verified
raw
history blame contribute delete
388 Bytes
import os
# Assumption that this secret exists in the current Space's environment
# You may choose between different models based on availability or requirements
def load_huggingface_model():
my_id = os.getenv("MODEL_URI")
# Your Hugging Face model loading logic here
prefix = "https://"
if not prefix in my_id:
my_id = prefix + my_id
return my_id