MSU576 commited on
Commit
c9481e0
·
verified ·
1 Parent(s): 72a027e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="/content/drive/MyDrive/soil_best_model.pth"):
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)