kishoreb4 commited on
Commit
564c9e3
·
verified ·
1 Parent(s): cb7125b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ NUM_CLASSES = len(ID2LABEL)
32
  def download_model_from_drive():
33
  # Create a models directory
34
  os.makedirs("models", exist_ok=True)
35
- model_path = "models/tf_model.h5"
36
 
37
  if not os.path.exists(model_path):
38
  # Fixed Google Drive URL format for gdown
 
32
  def download_model_from_drive():
33
  # Create a models directory
34
  os.makedirs("models", exist_ok=True)
35
+ model_path = "models/best_model"
36
 
37
  if not os.path.exists(model_path):
38
  # Fixed Google Drive URL format for gdown