MalikSahib1 commited on
Commit
9014494
·
verified ·
1 Parent(s): c52f854

Delete load_model.py

Browse files
Files changed (1) hide show
  1. 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.")