Update app.py

app.py

@@ -1,16 +1,5 @@
-import gradio as gr
-from transformers import pipeline
-import torch
-from openprompt.plms import load_plm
-from openprompt import PromptDataLoader
-from openprompt.prompts import ManualVerbalizer
-from openprompt.prompts import ManualTemplate
-from openprompt.data_utils import InputExample
-from openprompt import PromptForClassification
-
-
-def sentiment_analysis(sentence, template, positive, neutral, negative):
-    model_name = "CCCC/ARCH_tuned_bert" 
+def sentiment_analysis(sentence, template, model_name, positive, neutral, negative):
+    model_name = model_name
     template = template.replace('[SENTENCE]', '{"placeholder":"text_a"}')
     template = template.replace('[MASK]', '{"mask"}')
     classes = ['positive', 'neutral', 'negative']
@@ -67,10 +56,12 @@ def sentiment_analysis(sentence, template, positive, neutral, negative):
 demo = gr.Interface(fn=sentiment_analysis,
                     inputs = [gr.Textbox(placeholder="Enter sentence here.",label="sentence"),
                             gr.Textbox(placeholder="Your template must have a [SENTENCE] token and a [MASK] token.",label="template"),
+                            gr.Radio(choices=["ARCH_tuned_robert","FNCH_tuned_robert","AREN_tuned_robert","FNEN_tuned_robert","bert-base-uncased"], label="model choics"),
                             gr.Textbox(placeholder="Separate words with Spaces.",label="positive label words"),
                             gr.Textbox(placeholder="Separate words with Spaces.",label="neutral label words"),
                             gr.Textbox(placeholder="Separate words with Spaces.",label="negative label words")
                             ],
-                    outputs="text")
+                    outputs="text",
+        )
     
-demo.launch()  
+demo.launch()