Update app.py

app.py

@@ -11,15 +11,13 @@ def classify(text):
 
 def predict_sentiment(text):
     if model == "finiteautomata/bertweet-base-sentiment-analysis":
-        # pipe = pipeline("text-classification", model="finiteautomata/bertweet-base-sentiment-analysis")
-        # out = pipe(text, return_all_scores=True)
-        # return {pred["label"]: pred["score"] for pred in out[0]}
-        return {"cathf": 0.3, "doghf": 0.7}
+        pipe = pipeline("text-classification", model="finiteautomata/bertweet-base-sentiment-analysis")
+        out = pipe(text, return_all_scores=True)
+        return {pred["label"]: pred["score"] for pred in out[0]}
     elif model == "vader":
-        # nltk.download('vader_lexicon')
-        # sia = SentimentIntensityAnalyzer()
-        # return sia.polarity_scores(text)
-        return {"catv": 0.3, "dogv": 0.7}
+        nltk.download('vader_lexicon')
+        sia = SentimentIntensityAnalyzer()
+        return sia.polarity_scores(text)
         
 
 
@@ -37,15 +35,21 @@ with demo:
 
         with gr.TabItem("Multiple Inputs"):
             gr.Markdown("A more complex interface for sentiment analysis with multiple inputs, including a dropdown, and some examples")
-            demo = gr.Interface(
-                fn=predict_sentiment,
-                inputs=[
-                    gr.Textbox(),
-                    # gr.Dropdown(
-                    #     ["finiteautomata/bertweet-base-sentiment-analysis", "vader"], label="Model"
-                    # ),
+            interface = gr.Interface(
+                predict_sentiment,
+                [
+                    gr.Textbox(placeholder="Your text input"),
+                    gr.Dropdown(
+                        ["finiteautomata/bertweet-base-sentiment-analysis", "vader"], label="Model"
+                    ),
                 ],
-                outputs="label"
+                "text",
+                examples=[
+                    ["Happy smile", "vader"],
+                    ["Happy smile", "finiteautomata/bertweet-base-sentiment-analysis"],
+                    ["Sad frown", "vader"],
+                    ["Sad frown", "finiteautomata/bertweet-base-sentiment-analysis"],
+                ]
             )
 
 demo.launch()