Update app.py
Browse files
app.py
CHANGED
|
@@ -232,7 +232,7 @@ import streamlit as st
|
|
| 232 |
# Load Soil Model (6 Classes)
|
| 233 |
# ----------------------------
|
| 234 |
@st.cache_resource
|
| 235 |
-
def load_soil_model(path="
|
| 236 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 237 |
try:
|
| 238 |
model = models.resnet18(pretrained=False)
|
|
|
|
| 232 |
# Load Soil Model (6 Classes)
|
| 233 |
# ----------------------------
|
| 234 |
@st.cache_resource
|
| 235 |
+
def load_soil_model(path="soil_best_model.pth"):
|
| 236 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 237 |
try:
|
| 238 |
model = models.resnet18(pretrained=False)
|