Hugging Face's logo

Spaces:
asadsandhu
/
RAGnosis
Sleeping

App Files Community
asadsandhu commited on
Commit
a174361
·
1 Parent(s): 8329c82

Finalized.

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +3 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import pandas as pd
3
  import faiss
4
  import numpy as np
5
  import torch
6
- from transformers import AutoTokenizer, AutoModelForCausalLM
7
  from sentence_transformers import SentenceTransformer
8
 
9
  # ----------------------
 
3
  import faiss
4
  import numpy as np
5
  import torch
6
+ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
7
  from sentence_transformers import SentenceTransformer
8
 
9
  # ----------------------
requirements.txt CHANGED
@@ -3,5 +3,6 @@ sentence-transformers
3
  faiss-cpu
4
  torch
5
  gradio
6
- sentencepiece
7
- accelerate
 
 
3
  faiss-cpu
4
  torch
5
  gradio
6
+ accelerate
7
+ bitsandbytes
8
+ sentencepiece