Neon-AI commited on
Commit
45ba9ef
·
verified ·
1 Parent(s): 301cc01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -32,12 +32,11 @@ def load_model():
32
  device_map=None
33
  )
34
 
 
35
  try:
36
  model = PeftModel.from_pretrained(base_model, MODEL_ID)
37
- print("✅ LoRA adapter loaded")
38
  except Exception:
39
- model = base_model
40
- print("ℹ️ No LoRA adapter found, using base model")
41
 
42
  model.to("cpu")
43
  model.eval()
 
32
  device_map=None
33
  )
34
 
35
+ # If LoRA adapters exist, this will load them
36
  try:
37
  model = PeftModel.from_pretrained(base_model, MODEL_ID)
 
38
  except Exception:
39
+ model = base_model # fallback if no LoRA
 
40
 
41
  model.to("cpu")
42
  model.eval()