Update app.py

app.py

@@ -1,4 +1,3 @@
-!pip install bitsandbytes accelerate
 import os
 from faiss import write_index
 import gradio as gr
@@ -9,9 +8,17 @@ from torch.utils.data import DataLoader, Dataset
 from datasets import load_dataset
 import pandas as pd
 import faiss
-from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, AutoModel
-from transformers import TextIteratorStreamer
+from transformers import (
+    AutoTokenizer, 
+    AutoModelForCausalLM, 
+    BitsAndBytesConfig, 
+    AutoModel, 
+    TextIteratorStreamer
+)
 from threading import Thread
+from transformers import TextIteratorStreamer
+
+
 
 torch.set_num_threads(2)
 HF_TOKEN = os.environ.get("SECRET_TOKEN")