ContentAgent / agents /model.py
yetessam's picture
Update agents/model.py
b22b9ee verified
raw
history blame
405 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")
print(my_id)
# Your Hugging Face model loading logic here
prefix = "https://"
if not prefix in my_id:
my_id = prefix + my_id
return my_id