Hugging Face's logo

Spaces:
BerserkerMother
/
Aftrhour_demo
Runtime error

App Files Community
BerserkerMother commited on
Commit
3bea6f8
·
1 Parent(s): 2f42358

Changes ner model to roberta

Browse files
Files changed (1) hide show
  1. elise/src/app.py +3 -1
elise/src/app.py CHANGED
@@ -10,7 +10,9 @@ from utils import df_to_json
10
 
11
 
12
  # prep models
13
- MODEL_CHECKPOINT = "BerserkerMother/restaurant_ner"
 
 
14
  parser = SentenceParser.from_huggingface(MODEL_CHECKPOINT)
15
 
16
  embedder = SentenceTransformer("all-MiniLM-L6-v2")
 
10
 
11
 
12
  # prep models
13
+ MODEL_CHECKPOINT = (
14
+ "tner/roberta-large-mit-restaurant" # "BerserkerMother/restaurant_ner"
15
+ )
16
  parser = SentenceParser.from_huggingface(MODEL_CHECKPOINT)
17
 
18
  embedder = SentenceTransformer("all-MiniLM-L6-v2")