Testing trained model

Author: Dilshan-Nuwantha

GlueFactory_Training.ipynb

@@ -1243,16 +1243,22 @@
    "source": [
     "# Upload your own images\n",
     "from google.colab import files\n",
+    "import os\n",
     "\n",
     "print(\"Upload first image:\")\n",
     "uploaded1 = files.upload()\n",
     "img1_name = list(uploaded1.keys())[0]\n",
+    "img1_path = os.path.join('/content', img1_name)\n",
     "\n",
     "print(\"\\nUpload second image:\")\n",
     "uploaded2 = files.upload()\n",
     "img2_name = list(uploaded2.keys())[0]\n",
     "\n",
-    "print(f\"\\n✅ Images uploaded: {img1_name}, {img2_name}\")"
+    "img2_path = os.path.join('/content', img2_name)\n",
+    "\n",
+    "print(f\"   Image 2: {img2_path} (exists: {os.path.exists(img2_path)})\")\n",
+    "print(f\"   Image 1: {img1_path} (exists: {os.path.exists(img1_path)})\")\n",
+    "print(f\"\\n✅ Images uploaded:\")"
    ]
   },
   {
@@ -1264,8 +1270,8 @@
    "source": [
     "# Match your uploaded images\n",
     "with torch.no_grad():\n",
-    "    data0 = load_and_preprocess_image(img1_name)\n",
-    "    data1 = load_and_preprocess_image(img2_name)\n",
+    "    data0 = load_and_preprocess_image(img1_path)\n",
+    "    data1 = load_and_preprocess_image(img2_path)\n",
     "    \n",
     "    batch = {\n",
     "        'view0': {'image': data0['image']},\n",