TropicalBee commited on
Commit
dda8a6e
·
verified ·
1 Parent(s): 22ab02f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ weights_loaded = False
15
  def load_model_weights():
16
  global weights_loaded, model
17
  if not weights_loaded:
18
- print(" Loading model weights into memory...")
19
  # Load weights from the safetensors file on CPU
20
  state_dict = load_file("model.safetensors", device="cpu")
21
 
@@ -25,7 +25,7 @@ def load_model_weights():
25
  model.load_state_dict(state_dict)
26
  model.eval()
27
  weights_loaded = True
28
- print(" Weights loaded successfully!")
29
 
30
  # 2. Define the strict patch transform (No randomness!)
31
  test_transform = transforms.Compose([
 
15
  def load_model_weights():
16
  global weights_loaded, model
17
  if not weights_loaded:
18
+ print(" Loading model weights into memory...")
19
  # Load weights from the safetensors file on CPU
20
  state_dict = load_file("model.safetensors", device="cpu")
21
 
 
25
  model.load_state_dict(state_dict)
26
  model.eval()
27
  weights_loaded = True
28
+ print(" Weights loaded successfully!")
29
 
30
  # 2. Define the strict patch transform (No randomness!)
31
  test_transform = transforms.Compose([