fix: pred time

app.py

@@ -2,6 +2,7 @@
 import torch
 import gradio as gr
 from model import SentimentAnalysisModel
+from timeit import default_timer as timer
 
 # Load the pre-trained sentiment analysis model
 model = SentimentAnalysisModel(bert_model_name="SamLowe/roberta-base-go_emotions", num_labels=7)
@@ -22,6 +23,7 @@ emoji_to_emotion = {
 
 # Function to make predictions
 def predict_sentiment(text):
+    start_time = timer()
     inputs = model.tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     input_ids = inputs["input_ids"]
     attention_mask = inputs["attention_mask"]
@@ -31,11 +33,12 @@ def predict_sentiment(text):
 
     logits = outputs.logits
     _, predicted_class = torch.max(logits, dim=1)
+    pred_time = round(timer() - start_time, 5)
     
     # Map predicted class to emoji
     result = emoji_to_emotion[predicted_class.item()]
     
-    return result
+    return result,pred_time
 
 # Create title, description and article strings
 title = "Emoji-aware Sentiment Analysis using Roberta Model"