Spaces:
Sleeping
Sleeping
Tom commited on
Commit ·
e39c55a
1
Parent(s): 2a1d898
fixed a path issue
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def predict(image_path, model, class_names):
|
|
| 62 |
return full_name, confidence
|
| 63 |
|
| 64 |
# Load the model
|
| 65 |
-
model_path = "
|
| 66 |
class_names = ['AC','As','Cb','Cc','Ci','Cs','Ct','Cu','Ns','Sc','St'] # Replace with your actual class names
|
| 67 |
model = load_model(model_path, num_classes=len(class_names))
|
| 68 |
|
|
|
|
| 62 |
return full_name, confidence
|
| 63 |
|
| 64 |
# Load the model
|
| 65 |
+
model_path = "VisionTransformer_with_crop_final_model.pth" # Replace with your model path
|
| 66 |
class_names = ['AC','As','Cb','Cc','Ci','Cs','Ct','Cu','Ns','Sc','St'] # Replace with your actual class names
|
| 67 |
model = load_model(model_path, num_classes=len(class_names))
|
| 68 |
|