remove arguana

app.py

@@ -33,7 +33,7 @@ corpus_lookups = {}
 queries = {}
 q_lookups = {}
 qrels = {}
-datasets = ["scifact", "arguana"]
+datasets = ["scifact"] # others are too large for the Space unfortunately :( 
 current_dataset = "scifact"
 
 def pool(last_hidden_states, attention_mask):
@@ -71,7 +71,6 @@ def load_model():
     model = PeftModel.from_pretrained(base_model_instance, CUR_MODEL)
     model = model.merge_and_unload()
     model.eval()
-    model.cuda()
 
 def load_corpus_embeddings(dataset_name):
     global retrievers, corpus_lookups
@@ -130,7 +129,10 @@ def encode_queries(dataset_name, postfix):
             embeds = pool(outputs.last_hidden_state, batch_dict['attention_mask'])
             embeds = F.normalize(embeds, p=2, dim=-1)
             encoded_embeds.append(embeds.cpu().numpy())
-    
+
+
+    # remove model from GPU
+    model = model.cpu()    
     return np.concatenate(encoded_embeds, axis=0)
 
 def search_queries(dataset_name, q_reps, depth=1000):
@@ -192,7 +194,7 @@ iface = gr.Interface(
     ],
     outputs=gr.JSON(label="Evaluation Results"),
     title="Promptriever Demo",
-    description="Select a dataset and enter a postfix prompt to evaluate the model's performance. Note: it takes about **ten seconds** for each dataset."
+    description="Select a dataset and enter a prompt to evaluate the model's performance. Note: it takes about **ten seconds** to evaluate.",
 )
 
 # Launch the interface

arguana/corpus_emb.0.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:21742104cd3b5ff805fe1a7432c960b6933159c1092d49f5c4cad74922916a9b
-size 35619068

arguana/corpus_emb.1.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:37b86d66f10459bc2c032741f70e9f1edba68b2b3fbf9c7b1c1bb6f6139fef02
-size 35619074

arguana/corpus_emb.2.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f0ebf48b7b6ad18402a90da0ca92b71128b43d81a85fdb425edcaa3767f213d1
-size 35602692

arguana/corpus_emb.3.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:82bb37e7f6d7e0728da781aaccaad49830708cd44abc455f8c3a81db1e4b4b0f
-size 35602679