madoss commited on
Commit
0d70765
1 Parent(s): 9a26bab

Update app.py

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