paths edited

app.py

@@ -7,11 +7,8 @@ import torchvision.transforms as T
 import gradio as gr
 from gradio import themes
 
-# Current path
-current_path = os.getcwd()
-
 # Define the model path
-model_path = os.path.join(current_path, "ml", "models", "model.pt")
+model_path = os.path.join("ml", "models", "model.pt")
 
 # Determine the device to use (GPU if available, otherwise CPU)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -59,8 +56,8 @@ def predict(image: Image.Image):
 
 
 examples = [
-    os.path.join(current_path, "ml", "data", "train_11890.jpg"),
-    os.path.join(current_path, "ml", "data", "train_11716.jpg"),
+    "ml/data/train_11890.jpg",
+    "ml/data/train_11716.jpg",
 ]
 
 theme = gr.Theme(