chore

Signed-off-by: Suvaditya Mukherjee <[email protected]>

app.py

@@ -135,24 +135,37 @@ if __name__ == "__main__":
         parallel=0,
     )
 
-    wandb.login(
-        key=os.getenv("WANDB_API_KEY")
-    )
+    # wandb.login(
+    #     key=os.getenv("WANDB_API_KEY")
+    # )
 
     model_name = "Qwen/Qwen2.5-3B-Instruct"
 
-    wandb.init(
-        project="resume-rag", 
-        name="zerogpu-run",
-        save_code=True,
-        config={
-            "model_name": model_name,
-            "resume_url": RESUME_URL
-        }
-    )
+    # wandb.init(
+    #     project="resume-rag", 
+    #     name="zerogpu-run",
+    #     save_code=True,
+    #     config={
+    #         "model_name": model_name,
+    #         "resume_url": RESUME_URL
+    #     }
+    # )
 
     @spaces.GPU
     def rag_process(message, chat_history):
+        wandb.login(
+            key=os.getenv("WANDB_API_KEY")
+        )
+
+        wandb.init(
+            project="resume-rag", 
+            name="zerogpu-run",
+            save_code=True,
+            config={
+                "model_name": model_name,
+                "resume_url": RESUME_URL
+            }
+        )
         # Append current user message to chat history
         current_message = {
             "role": "user",
@@ -195,8 +208,6 @@ if __name__ == "__main__":
         }
         wandb.log(metrics)
 
-        wandb.finish()
-
         return generated_text[:-10]
 
     model = AutoModelForCausalLM.from_pretrained(
@@ -221,7 +232,8 @@ if __name__ == "__main__":
         title="Suvaditya's Personal RAG, a space on ZeroGPU!",
         examples=["Where did Suvaditya complete his Bachelor's Degree?", "Where is Suvaditya currently working?"],
         description="Ask any question about Suvaditya's resume and get an answer!",
-        # theme="glass"
         theme="John6666/YntecDark"
     )
-    demo.launch()
+    demo.launch()
+
+    wandb.finish()