updates

Author: AseelOmer

google-gemma/gemma.ipynb

@@ -258,7 +258,7 @@
    ],
    "source": [
     "# Cell 1\n",
-    "!pip install --upgrade huggingface_hub sentence-transformers codecarbon pypdf torch scikit-learn pandas\n"
+    "!pip install --upgrade huggingface_hub sentence-transformers codecarbon pypdf torch scikit-learn pandas"
    ]
   },
   {
@@ -295,7 +295,6 @@
     "import time\n",
     "import textwrap\n",
     "import glob\n",
-    "import pandas as pd\n",
     "import numpy as np\n",
     "from huggingface_hub import InferenceClient\n",
     "from sentence_transformers import SentenceTransformer\n",
@@ -322,7 +321,7 @@
     "print(\"Data folder:\", DATA_PATH)\n",
     "print(\"Answer log file:\", LOG_FILE)\n",
     "print(\"Emissions log file:\", EMISSIONS_FILE)\n",
-    "print(\"API key loaded?\", (HF_API_KEY[:10] + \"...\") if HF_API_KEY else \"MISSING\")\n"
+    "print(\"API key loaded?\", (HF_API_KEY[:10] + \"...\") if HF_API_KEY else \"MISSING\")"
    ]
   },
   {
@@ -369,7 +368,7 @@
     ")\n",
     "REFINER_USER_TEMPLATE = \"Original Draft:\\n{draft}\\n\\nCritic's Feedback:\\n{feedback}\"\n",
     "\n",
-    "print(\"Prompts ready to use.\")\n"
+    "print(\"Prompts ready to use.\")"
    ]
   },
   {
@@ -424,6 +423,7 @@
    "source": [
     "# Cell 4\n",
     "\n",
+    "\n",
     "class Node:\n",
     "    def __init__(self, content, doc_id, chunk_id, fname=None):\n",
     "        self._content = content\n",
@@ -494,6 +494,8 @@
     "    sims = embeddings_norm @ q_emb\n",
     "    topk_idx = np.argsort(-sims)[:k]\n",
     "    return [nodes[i] for i in topk_idx]\n",
+    "\n",
+    "\n",
     "print(\"Retriever function ready to use.\")"
    ]
   },
@@ -573,7 +575,7 @@
     "print(\"\\nRunning a tiny API test...\")\n",
     "test_prompt = \"Who was the first person on the Moon?\"\n",
     "resp = call_hf_api(\"You are a helpful assistant.\", test_prompt, max_tokens=60)\n",
-    "print(\"\\n--- Test response ---\\n\", resp)\n"
+    "print(\"\\n--- Test response ---\\n\", resp)"
    ]
   },
   {
@@ -2060,7 +2062,7 @@
     "    total_emissions = tracker.stop()\n",
     "    print(\"\\n Total Emissions Summary:\")\n",
     "    print(f\"{total_emissions * 1000:.6f} gCO2eq\")\n",
-    "    print(\"Notebook finished.\")\n"
+    "    print(\"Notebook finished.\")"
    ]
   }
  ],