Files changed (1) hide show
  1. app.py +3 -0
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