Update roberta_predict.py

roberta_predict.py

@@ -9,11 +9,12 @@ MAX_LEN = 64
 labels = ["Negative", "Neutral", "Positive"]
 
 
+
 MODEL_REPOS = {
-    "roberta": "subhankarmannayfy/brand-roberta",
-    "distilroberta": "subhankarmannayfy/brand-distilroberta",
-    "bert": "subhankarmannayfy/brand-bert",
-    "albert": "subhankarmannayfy/brand-albert"
+    "roberta":       os.getenv("ROBERTA_MODEL"),
+    "distilroberta": os.getenv("DISTILROBERTA_MODEL"),
+    "bert":          os.getenv("BERT_MODEL"),
+    "albert":        os.getenv("ALBERT_MODEL"),
 }