madoss commited on
Commit
3a29d4a
·
1 Parent(s): d82542d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,11 +3,12 @@ import logging
3
 
4
  import datasets
5
  import gradio as gr
 
6
 
7
  logging.disable(logging.CRITICAL)
8
 
9
- model_name = "dangvantuan/sentence-camembert-large"
10
- model = gr.load(model_name)
11
 
12
  dataset = datasets.load_dataset("json", data_files=["./data/dataset.json"], split="train")
13
  dataset.load_faiss_index("embeddings", "index.faiss")
 
3
 
4
  import datasets
5
  import gradio as gr
6
+ import sentence_transformers
7
 
8
  logging.disable(logging.CRITICAL)
9
 
10
+ model = sentence_transformers.SentenceTransformer(
11
+ "dangvantuan/sentence-camembert-large", device="cpu")
12
 
13
  dataset = datasets.load_dataset("json", data_files=["./data/dataset.json"], split="train")
14
  dataset.load_faiss_index("embeddings", "index.faiss")