Updated workflow to include milestone-3

.github/workflows/sync_to_huggingface.yaml

@@ -1,7 +1,7 @@
 name: Sync to Hugging Face Hub
 on:
   push:
-    branches: [milestone-2]
+    branches: [milestone-3]
   
   workflow_dispatch:
 
@@ -16,4 +16,4 @@ jobs:
       - name: Push to hub
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
-        run: git push https://MJ3128:[email protected]/spaces/MJ3128/CS-GY-6613-Project milestone-2:main
+        run: git push https://MJ3128:[email protected]/spaces/MJ3128/CS-GY-6613-Project milestone-3:main

app.py

@@ -8,7 +8,7 @@ if "score" not in st.session_state:
     st.session_state.score = ""
 
 
-def run_model(text_in, model_in):
+def run_sentiment_model(text_in, model_in):
     classifier = pipeline(task="sentiment-analysis",
                           model=model_in)
     analysis = classifier(text_in)
@@ -20,14 +20,16 @@ models_available = {"Roberta Large English": "siebert/sentiment-roberta-large-en
                     "Generic": "Seethal/sentiment_analysis_generic_dataset",
                     "Twitter Roberta": "cardiffnlp/twitter-roberta-base-sentiment"}
 
-st.title("Sentiment Analysis Web Application")
+st.title("Sentiment Analysis Section (Milestone-2)")
 text_input = st.text_area(
     label="Enter the text to analyze", value="I Love Pizza")
 model_picked = st.selectbox(
     "Choose a model to run on", options=models_available.keys())
 
-st.button("Submit", on_click=run_model, args=(
+st.button("Submit", on_click=run_sentiment_model, args=(
     text_input, models_available[model_picked]))
 
 st.markdown(body="Sentiment: {}, Confidence Score: {} %".format(
     st.session_state.sentiment, st.session_state.score))
+
+st.title("Patentability Score Section (Milestone-3)")