Update app.py (#1)

- Update app.py (3923a5c74fd0f3d20d79b96ddaa1897cd022ac62)


Co-authored-by: shecodes <shecodesinsg@users.noreply.huggingface.co>

app.py

@@ -1,4 +1,3 @@
-
 import pandas as pd
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import random
@@ -10,13 +9,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
 
 # Path to your model file (adjust this if needed)
-model_path = "bhadresh-savani/bert-base-uncased-emotion"
-
-#my code
-# Load pre-trained model and tokenizer
 model_name = "bhadresh-savani/bert-base-uncased-emotion"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
 def predict_emotion(text):
     # Tokenize the input text
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
@@ -46,50 +42,36 @@ def predict_emotion(text):
 motivational_quotes = {
     "anger": [
         "It’s okay to feel angry when things don’t go as expected. Your emotions are valid and show you care deeply. Take a moment to breathe and channel that energy into finding a solution—you’ve got the strength to turn this around!"
-          ],
+    ],
     "joy": [
         "Your joy is contagious, and it’s wonderful to see you so happy! Celebrate this moment fully, and let it inspire you to keep reaching for more moments like these."
     ],
     "optimism": [
         "Your positive outlook is inspiring! Keep holding onto that hope and belief in yourself—it will guide you through any challenge. The best is yet to come!"
-        ],
+    ],
     "sadness": [
         "It’s okay to feel sad—allow yourself to process and heal. Remember, tough times don’t last forever. Lean on those who care about you, and know that brighter days are ahead."
     ],
     "fear": [
         "Feeling afraid is natural; it shows you’re stepping outside your comfort zone. Trust in your abilities and take things one step at a time. You’re braver than you think!"
-        ],
-    "surprise" : [
+    ],
+    "surprise": [
         "Life’s surprises can be startling or exciting, but they often bring new opportunities. Embrace the unexpected with an open mind—you might discover something amazing."
-         ]
-
+    ]
 }
 
-#function to generate motivational quotes
+# Function to generate motivational quotes
 def get_motivational_quote(emotion):
     if emotion in motivational_quotes:
         return random.choice(motivational_quotes[emotion])
     else:
         return "Stay strong and keep moving forward!"
 
-date = input("Enter date of entry (DD/MM/YYYY): ")
-entry = input("Enter your feelings & thoughts: ")
-
-emotion, confidence = predict_emotion(entry)
-quote = get_motivational_quote(emotion)
-
-print(f"\nOn {date}, your detected emotion is: {emotion} (Confidence: {confidence:.2f})")
-print(f"Here's something to tell yourself: {quote}")
-print(f"No matter what you’re feeling, today is a fresh start —- embrace it with courage, knowing you have the strength to handle anything that comes your way! Thanks for sharing your feelings!", end='')
-#end of my code
-
-
 def process_entry(date, text):
     emotion, confidence = predict_emotion(text)
     quote = get_motivational_quote(emotion)
     return f"On {date}, your detected emotion is: {emotion} (Confidence: {confidence:.2f})", quote
 
-
 interface = gr.Interface(
     fn=process_entry,
     inputs=[