DrishtiSharma commited on
Commit
a0e8323
1 Parent(s): 88b492c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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