incorporated tweetnlp

README.md

@@ -1,6 +1,6 @@
 ---
 title: Twitter Sentiment
-emoji: 👀
+emoji: 🐦
 colorFrom: blue
 colorTo: red
 sdk: gradio

app.py

@@ -1,7 +1,24 @@
 import gradio as gr
+from tweetnlp import Sentiment
 
-def greet(name):
-    return "Hello " + name + "!!"
+def classify(tweet):
+    # return "Hello " + name + "!!"
+    model_output = model.sentiment(tweet)
+    # Fill in both positive and negative values
+    if model_output["label"] == "positive":
+        formatted_output = dict()
+        formatted_output["positive"] = model_output["probability"]
+        formatted_output["negative"] = 1 - model_output["probability"]
+    else:
+        formatted_output = dict()
+        formatted_output["negative"] = model_output["probability"]
+        formatted_output["positive"] = 1 - model_output["probability"]
+    return formatted_output
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+
+if __name__ == "__main__":
+    # https://github.com/cardiffnlp/tweetnlp
+    model = Sentiment()
+    iface = gr.Interface(fn=classify, inputs="text", outputs="label")
+    iface.launch()

py_env.yaml

@@ -0,0 +1,15 @@
+# # Create conda environemnt
+# conda env create -f py_env.yaml
+
+# # Update Conda environement
+# conda env update -f py_env.yaml  --prune
+
+name: twitter_env
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - pip
+  - pip:
+    - gradio
+    - tweetnlp