Spaces:
Sleeping
Sleeping
Update app.py (#29)
Browse files- Update app.py (c8b63b6163fe45c44b3b7581735183cd940f8798)
app.py
CHANGED
@@ -8,7 +8,7 @@ from typing import Union
|
|
8 |
|
9 |
def download_file(media_type, digest, image):
|
10 |
url = f"https://registry.ollama.ai/v2/library/{image}/blobs/{digest}"
|
11 |
-
f_tag =
|
12 |
if f_tag == "model":
|
13 |
f_tag = "model.gguf"
|
14 |
file_name = f"blobs/{f_tag}"
|
|
|
8 |
|
9 |
def download_file(media_type, digest, image):
|
10 |
url = f"https://registry.ollama.ai/v2/library/{image}/blobs/{digest}"
|
11 |
+
f_tag = media_type.split('.')[-1]
|
12 |
if f_tag == "model":
|
13 |
f_tag = "model.gguf"
|
14 |
file_name = f"blobs/{f_tag}"
|