lukmanaj commited on
Commit
5783d78
·
verified ·
1 Parent(s): 84e454c

Rename streamlit-app.py to app.py

Browse files
Files changed (1) hide show
  1. streamlit-app.py → app.py +1 -1
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 = 'model/densenetafri.pth' # Fixed 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