Update app.py

app.py

@@ -18,9 +18,22 @@ import streamlit as st
 
 # Prog3: Zero Shot Classifier 
 
-classifier2 = pipeline("zero-shot-classification")
-res = classifier2(
-    "This is a course for Python List comprehension",
-    candidate_labels = ["education","politics","business"]
-)
-st.write(res)
+# classifier2 = pipeline("zero-shot-classification")
+# res = classifier2(
+#     "This is a course for Python List comprehension",
+#     candidate_labels = ["education","politics","business"]
+# )
+# st.write(res)
+
+
+# Prog4: Using Automodel and Autotokenizer
+
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+classifier3 = pipeline("sentiment-analysis",model=model,tokenizer=tokenizer)
+
+res = classifier3("Newton has been the biggest physicist")