Spaces:

bgk
/

sipariseng

Sleeping

bgk commited on Aug 1, 2023

Commit

7211c44

1 Parent(s): 9181316

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ labels = ["O", "B-FOOD_QUANTITY", "B-FOOD_SIZE", "B-FOOD", "I-FOOD", "B-FOOD_ING
 model = NERModel(
         "roberta",
-        "bgk/mobilebert", labels=labels,
         use_cuda=False,
         ignore_mismatched_sizes=True
     )

 model = NERModel(
         "roberta",
+        "bgk/berteng", labels=labels,
         use_cuda=False,
         ignore_mismatched_sizes=True
     )