Spaces:

BlueSR
/

Road_Surface_Classification

Sleeping

BlueSR commited on Jun 5, 2025

Commit

4014e2e

verified ·

1 Parent(s): 99be414

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,6 @@ from torchvision import transforms
 from torchvision.models import efficientnet_b0
 import gradio as gr
 import numpy as np
-import pickle
-import os
 from PIL import Image
 import joblib
 import json
@@ -14,7 +12,7 @@ from sklearn.tree import export_text
 # Constants
 NUM_CLASSES = 4
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-CLASS_NAMES = ["Good_condition", "Potholes", "Cracked_road", "Flooded_muddy"]
 # Load the CNN model
 def load_cnn_model():

 from torchvision.models import efficientnet_b0
 import gradio as gr
 import numpy as np
 from PIL import Image
 import joblib
 import json
 # Constants
 NUM_CLASSES = 4
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+CLASS_NAMES = ['Cracked_road', 'Flooded_muddy', 'Good_condition', 'Potholes']
 # Load the CNN model
 def load_cnn_model():