ContentAgent / agents /model.py
yetessam's picture
Rename to huggingface_model_uri
0c0b4ea verified
raw
history blame
404 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 huggingface_model_uri():
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