Update app.py

app.py

@@ -8,7 +8,10 @@ from typing import Union
 
 def download_file(media_type, digest, image):
     url = f"https://registry.ollama.ai/v2/library/{image}/blobs/{digest}"
-    file_name = f"blobs/{media_type.split('.')[-1]}"
+    f_tag = {media_type.split('.')[-1]}
+    if f_tag == "model":
+        f_tag = "model.gguf"
+    file_name = f"blobs/{f_tag}"
 
     # Create the directory if it doesn't exist
     os.makedirs(os.path.dirname(file_name), exist_ok=True)