Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
#46
by
cappuch
- opened
app.py
CHANGED
@@ -121,6 +121,9 @@ def download_from_url(url, name=None):
|
|
121 |
print(pth_file)
|
122 |
print(index_file)
|
123 |
|
|
|
|
|
|
|
124 |
MODELS.append({"model": pth_file, "index": index_file, "model_name": name})
|
125 |
return ["Downloaded as " + name, pth_files[0], index_files[0]]
|
126 |
|
|
|
121 |
print(pth_file)
|
122 |
print(index_file)
|
123 |
|
124 |
+
if name == "":
|
125 |
+
name = pth_file.split(".")[0]
|
126 |
+
|
127 |
MODELS.append({"model": pth_file, "index": index_file, "model_name": name})
|
128 |
return ["Downloaded as " + name, pth_files[0], index_files[0]]
|
129 |
|