KeerthiVM commited on
Commit
1b1c00e
·
1 Parent(s): a0c0f50

debugging fix

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -371,9 +371,18 @@ def initialize_classifier():
371
  print("⚙️ Initializing skin disease classifier...")
372
  classifier = SkinDiseaseClassifier()
373
  classifier.load_models()
 
 
 
 
374
  return classifier
375
 
376
- classifier = initialize_classifier()
 
 
 
 
 
377
 
378
  # === Session Init ===
379
  if "messages" not in st.session_state:
 
371
  print("⚙️ Initializing skin disease classifier...")
372
  classifier = SkinDiseaseClassifier()
373
  classifier.load_models()
374
+ dummy_img = Image.new('RGB', (224, 224))
375
+ classifier.predict(dummy_img)
376
+
377
+ print("⚙️ Initialization successful")
378
  return classifier
379
 
380
+ # print("⚙️ Initializing skin disease classifier...")
381
+ # classifier = initialize_classifier()
382
+ # print("⚙️ Initializing classifier done...")
383
+ with st.spinner("Loading AI models (one-time operation)..."):
384
+ classifier = initialize_classifier()
385
+ st.success("Models loaded successfully!")
386
 
387
  # === Session Init ===
388
  if "messages" not in st.session_state: