Change model_path, base_data_path

app.py

@@ -394,8 +394,8 @@ if __name__ == "__main__":
     batch_size = 48
 
     # Paths
-    base_data_path = f"Data/Data_{patch_size}_{version}"
-    model_path = f"Models/model__{patch_size}_{batch_size}_{version}.pth"
+    base_data_path = "Data/hf_Data_{patch_size}_{version}"
+    model_path = "model__{patch_size}_{batch_size}_{version}.pth"
 
     # Create and launch interface
     interface = create_gradio_interface(