Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -177,6 +177,8 @@ def get_huggingface_safetensors(link):
|
|
177 |
if len(split_link) != 2:
|
178 |
raise Exception("Invalid Hugging Face repository link format.")
|
179 |
|
|
|
|
|
180 |
# Load model card
|
181 |
model_card = ModelCard.load(link)
|
182 |
base_model = model_card.data.get("base_model")
|
@@ -242,6 +244,7 @@ def get_huggingface_safetensors(link):
|
|
242 |
|
243 |
|
244 |
def check_custom_model(link):
|
|
|
245 |
if(link.startswith("https://")):
|
246 |
if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
|
247 |
link_split = link.split("huggingface.co/")
|
|
|
177 |
if len(split_link) != 2:
|
178 |
raise Exception("Invalid Hugging Face repository link format.")
|
179 |
|
180 |
+
print(f"Repository attempted: {split_link}")
|
181 |
+
|
182 |
# Load model card
|
183 |
model_card = ModelCard.load(link)
|
184 |
base_model = model_card.data.get("base_model")
|
|
|
244 |
|
245 |
|
246 |
def check_custom_model(link):
|
247 |
+
print(f"Checking a custom model on: {link}")
|
248 |
if(link.startswith("https://")):
|
249 |
if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
|
250 |
link_split = link.split("huggingface.co/")
|