File size: 388 Bytes
0e2f732
 
f8c087a
0e2f732
 
 
 
116fe2e
 
f8c087a
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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