Spaces:

pahri
/

setfit-absa-resto-demo

Sleeping

pahri commited on May 15

Commit

0088ba1

•

1 Parent(s): b4b0220

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from setfit import AbsaModel
 # Load the ABSA model (assuming indo-setfit-absa-bert-base-restaurants is the aspect extraction model)
 model = AbsaModel.from_pretrained(
-    "firqaaa/indo-setfit-absa-bert-base-restaurants-aspect",
-    "firqaaa/indo-setfit-absa-bert-base-restaurants-polarity",
     spacy_model="id_core_news_trf",
 )

 # Load the ABSA model (assuming indo-setfit-absa-bert-base-restaurants is the aspect extraction model)
 model = AbsaModel.from_pretrained(
+    "pahri/setfit-indo-resto-RM-ibu-imas-polarity",
+    "pahri/setfit-indo-resto-RM-ibu-imas-aspect",
     spacy_model="id_core_news_trf",
 )