Update app.py

app.py

@@ -100,6 +100,9 @@ def save_historical_data(historical_data, file_path=emotion_history_file):
 emotion_history = load_historical_data()
 
 def update_emotion(emotion, percentage, intensity):
+    if percentage > emotions['ideal_state']['percentage']:
+        percentage = emotions['ideal_state']['percentage']
+    
     emotions['ideal_state']['percentage'] -= percentage
     emotions[emotion]['percentage'] += percentage
     emotions[emotion]['intensity'] = intensity
@@ -178,7 +181,7 @@ def predict_emotion(context):
     tokens = emotion_prediction_tokenizer(context, return_tensors='pt', padding=True, truncation=True)
     outputs = emotion_prediction_model(**tokens)
     logits = outputs.logits
-    predicted_class = torch.argmax(logits, dim=1)
+    predicted_class = torch.argmax(logits, dim=1).item()
     predicted_emotion = emotion_classes[predicted_class]
 
     # Get percentage and intensity based on context
@@ -214,4 +217,4 @@ with gr.Blocks() as demo:
     submit.click(update_ui, user_input, response)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)