Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import json
+import requests
 
 negative = "(worst quality, low quality:1.4), (bad-image-v2-39000, bad_prompt_version2:0.6)"
 sampler = "DPM++ 2M Karras"
@@ -19,6 +20,16 @@ models = [
     "https://civitai.com/api/download/models/6792?type=VAE&format=Other"
 ]
 
+valid_extension = ["safetensors", "ckpt", "pt"]
+for i in range(models):
+    if models[i].split('.')[-1] not in valid_extension:
+        r = requests.head(models[i], allow_redirects=True)
+        models[i] = r.url
+    
+        models[i] = [models[i], requests.utils.unquote(r.url).split("filename=")[1].split('"')[1]]
+    else:
+        models[i] = [models[i], models[i].split('/')[-1]]
+        
 default_model = None
 default_vae = None