Spaces:
Sleeping
Sleeping
Rename streamlit-app.py to app.py
Browse files
streamlit-app.py → app.py
RENAMED
@@ -55,7 +55,7 @@ if uploaded_file is not None:
|
|
55 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
56 |
|
57 |
# Load the model
|
58 |
-
model_path = '
|
59 |
model = load_model(model_path, device)
|
60 |
|
61 |
# Preprocess the image and predict
|
|
|
55 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
56 |
|
57 |
# Load the model
|
58 |
+
model_path = 'densenetafri.pth' # Fixed model path
|
59 |
model = load_model(model_path, device)
|
60 |
|
61 |
# Preprocess the image and predict
|