first commit

app.py

@@ -38,8 +38,8 @@ if model_choice not in st.session_state.downloaded_models:
         st.toast(f"✅ {model_choice} Succesfuly Download!", icon="🎉")
 
 # **1. Sentiment Analysis Model**
-if model_choice == "TinyBERT Sentiment Analysis":
-    text = st.text_area("Enter Text:", "Your review...")
+if model_choice == "TinyBert Sentiment Analysis":
+    text = st.text_area("Enter Text:", "This movie was horrible, the plot was really boring. acting was okay")
     predict = st.button("Predict Sentiment")
 
     classifier = pipeline("text-classification", model=local_path, device=device)
@@ -51,7 +51,7 @@ if model_choice == "TinyBERT Sentiment Analysis":
 
 # **2. Disaster Classification**
 if model_choice == "TinyBert Disaster Classification":
-    text = st.text_area("Enter Text:", "Your Tweet...")
+    text = st.text_area("Enter Text:", "There is a fire in the building")
     predict = st.button("Predict Sentiment")
 
     classifier = pipeline("text-classification", model=local_path, device=device)
@@ -70,7 +70,7 @@ if model_choice == "VIT Pose Classification":
         image = Image.open(uploaded_file)
         st.image(image, caption="Your Image", use_column_width=True)
 
-        image_processor = AutoImageProcessor.from_pretrained(local_directory, use_fast=True)
+        image_processor = AutoImageProcessor.from_pretrained(local_path, use_fast=True)
         pipe = pipeline('image-classification', model=local_path, image_processor=image_processor, device=device)
 
         if predict: