Spaces:
Running
Running
debugging fix
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
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:
|