Hugging Face's logo

Spaces:
TheresaQWQ
/
jina-rerank
Sleeping

App Files Community
TheresaQWQ commited on
Commit
4bd5edf
·
verified ·
1 Parent(s): bd04035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -8,7 +8,6 @@ model = AutoModelForSequenceClassification.from_pretrained(
8
  trust_remote_code=True,
9
  )
10
 
11
- model.to('cuda') # or 'cpu' if no GPU is available
12
  model.eval()
13
 
14
  def compute_scores(query, documents):
 
8
  trust_remote_code=True,
9
  )
10
 
 
11
  model.eval()
12
 
13
  def compute_scores(query, documents):