Spaces:
Sleeping
Sleeping
Delete load_model.py
Browse files- load_model.py +0 -24
load_model.py
DELETED
@@ -1,24 +0,0 @@
|
|
1 |
-
# load_model.py
|
2 |
-
|
3 |
-
# Is file ka kaam sirf server ke start hone per
|
4 |
-
# zaroori models ko download karna hai taake woh cache mein aa jayein.
|
5 |
-
|
6 |
-
print("Pre-loading all required AI models...")
|
7 |
-
|
8 |
-
try:
|
9 |
-
print("Downloading 'rembg' model (u2net)...")
|
10 |
-
from rembg.bg import new_session
|
11 |
-
new_session("u2net")
|
12 |
-
print("'rembg' model downloaded successfully.")
|
13 |
-
except Exception as e:
|
14 |
-
print(f"Could not pre-load rembg model: {e}")
|
15 |
-
|
16 |
-
try:
|
17 |
-
print("Downloading 'diffusers' Tiny Stable Diffusion model...")
|
18 |
-
from diffusers import DiffusionPipeline
|
19 |
-
DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-stable-diffusion-torch")
|
20 |
-
print("'diffusers' model downloaded successfully.")
|
21 |
-
except Exception as e:
|
22 |
-
print(f"Could not pre-load diffusers model: {e}")
|
23 |
-
|
24 |
-
print("All models pre-loading finished.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|