Spaces:

dhe1raj
/

breedcow

Sleeping

dhe1raj commited on Sep 7

Commit

d6ec985

verified ·

1 Parent(s): 66d925a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 # CONFIGURATION
 # =======================
 device = "cuda" if torch.cuda.is_available() else "cpu"
-MODEL_PATH = "best_model.pth"  # Upload your .pth model here
 CLASS_NAMES = ["Gir", "Deoni", "Murrah"]
 # =======================

 # CONFIGURATION
 # =======================
 device = "cuda" if torch.cuda.is_available() else "cpu"
+MODEL_PATH = "cattle_breed_efficientnetb3_pytorch.pth"  # Upload your .pth model here
 CLASS_NAMES = ["Gir", "Deoni", "Murrah"]
 # =======================