RoBERTa

app.py

@@ -11,7 +11,7 @@ from model import FFNModule, FeatureNormPredictor, FFNParams, TrainingParams
 def predict (Sentence, Word, LLM, Norm, Layer):
     models = {'BERT': 'bert-base-uncased',
               'ALBERT': 'albert-xxlarge-v2',
-              'roBERTa': 'roberta-base'}
+              'RoBERTa': 'roberta-base'}
     if Word not in Sentence: return "invalid input: word not in sentence"
     model_name_hf = LLM.lower()
     norm_name_hf = Norm.lower()
@@ -52,7 +52,7 @@ demo = gr.Interface(
     inputs=[
         "text", 
         "text", 
-        gr.Radio(["BERT", "ALBERT", "roBERTa"]),
+        gr.Radio(["BERT", "ALBERT", "RoBERTa"]),
         gr.Radio(["Binder", "McRae", "Buchanan"]),
         gr.Slider(0,12, step=1)
     ],