o

inference.py

@@ -7,6 +7,31 @@ from seq2seq_inference import infer_t5_prompt, infer_mBart_prompt
 from peft import LoraConfig, get_peft_model, PeftModel
 from modeling_bilstm_crf import BERT_BiLSTM_CRF
 
+# Define supported models and their adapter IDs
+MODEL_OPTIONS = {
+
+    "Araberta": {
+        "base": "asmashayea/absa-araberta",
+        "adapter": "asmashayea/absa-araberta"
+    },
+    "mT5": {
+        "base": "google/mt5-base",
+        "adapter": "asmashayea/mt4-absa"
+    },
+    # "mBART": {
+    #     "base": "facebook/mbart-large-50-many-to-many-mmt",
+    #     "adapter": "asmashayea/mbart-absa"
+    # },
+    "GPT3.5": {
+        "base": "bigscience/bloom-560m",  # example, not ideal for ABSA
+        "adapter": "asmashayea/gpt-absa"
+    },
+    "GPT4o": {
+        "base": "bigscience/bloom-560m",  # example, not ideal for ABSA
+        "adapter": "asmashayea/gpt-absa"
+    }
+}
+
 cached_models = {}
 
 def load_araberta():