classes

app.py

@@ -24,7 +24,7 @@ STATIC_DIR.mkdir(parents=True, exist_ok=True)
 
 # Global variables
 MODEL_PATH = MODELS_DIR / "resnet_50.pth"
-CLASSES_PATH = MODELS_DIR / "classes.txt"
+CLASSES_PATH =  BASE_DIR / "classes.txt"
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 def load_class_labels() -> Optional[list]:
@@ -152,7 +152,7 @@ def get_example_list() -> list:
     try:
         examples = []
         for ext in ['.jpg', '.jpeg', '.png']:
-            examples.extend(list(EXAMPLES_DIR.glob(f'*{ext}')))
+            examples.extend(list(EXAMPLES_DIR.glob(f'*.{ext}')))
         return [[str(ex)] for ex in sorted(examples)]
     except Exception as e:
         logger.error(f"Error loading examples: {str(e)}")