DrishtiSharma
commited on
Commit
•
a0e8323
1
Parent(s):
88b492c
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ torch_device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
30 |
print("Running on device:", torch_device)
|
31 |
print("CPU threads:", torch.get_num_threads())
|
32 |
|
33 |
-
model_id = "NousResearch/Llama-2-7b-hf"
|
34 |
biencoder = SentenceTransformer("intfloat/e5-large-v2", device=torch_device)
|
35 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-12-v2", max_length=512, device=torch_device)
|
36 |
|
|
|
30 |
print("Running on device:", torch_device)
|
31 |
print("CPU threads:", torch.get_num_threads())
|
32 |
|
33 |
+
model_id = "NousResearch/Llama-2-7b-chat-hf"
|
34 |
biencoder = SentenceTransformer("intfloat/e5-large-v2", device=torch_device)
|
35 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-12-v2", max_length=512, device=torch_device)
|
36 |
|