Update app.py
Browse files
app.py
CHANGED
@@ -51,13 +51,6 @@ def load_model(
|
|
51 |
final_model_path = os.path.join(directory, model_name)
|
52 |
|
53 |
print("Downloading all files...")
|
54 |
-
rm_files = [os.path.join(directory, f) for f in os.listdir(directory)]
|
55 |
-
for f in rm_files:
|
56 |
-
if os.path.isfile(f):
|
57 |
-
os.remove(f)
|
58 |
-
else:
|
59 |
-
shutil.rmtree(f)
|
60 |
-
|
61 |
if not os.path.exists(final_model_path):
|
62 |
with open(final_model_path, "wb") as f:
|
63 |
http_get(model_url, f)
|
|
|
51 |
final_model_path = os.path.join(directory, model_name)
|
52 |
|
53 |
print("Downloading all files...")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
if not os.path.exists(final_model_path):
|
55 |
with open(final_model_path, "wb") as f:
|
56 |
http_get(model_url, f)
|