yetessam commited on
Commit
116fe2e
·
verified ·
1 Parent(s): e044b5c

Update agents/model.py

Browse files
Files changed (1) hide show
  1. agents/model.py +7 -1
agents/model.py CHANGED
@@ -4,4 +4,10 @@ import os
4
  def load_huggingface_model():
5
  my_id = os.getenv("MODEL_URI")
6
  # Your Hugging Face model loading logic here
7
- return my_id
 
 
 
 
 
 
 
4
  def load_huggingface_model():
5
  my_id = os.getenv("MODEL_URI")
6
  # Your Hugging Face model loading logic here
7
+ prefix = "https://"
8
+
9
+ if prefix in my_id:
10
+ return my_id
11
+ else:
12
+ print("Substring not found!")
13
+ return prefix + my_id