ContentAgent / agents /model.py
yetessam's picture
Update agents/model.py
116fe2e verified
raw
history blame
353 Bytes
import os
# 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 prefix in my_id:
return my_id
else:
print("Substring not found!")
return prefix + my_id