Update app.py

app.py

@@ -1,10 +1,10 @@
 # Core imports
-from huggingface_hub import login
 import os
 import logging
 import torch
 from typing import Dict, List, Any
 import gradio as gr
+from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from sentence_transformers import SentenceTransformer
 import faiss
@@ -13,6 +13,7 @@ from tqdm import tqdm
 from datetime import datetime
 from dataclasses import dataclass, field
 from dotenv import load_dotenv
+import wandb 
 
 # Load environment variables
 load_dotenv()